Searched refs:Attr (Results 1 - 25 of 457) sorted by relevance

1234567891011>>

/external/tensorflow/tensorflow/examples/adding_an_op/
H A Dattr_examples.cc19 REGISTER_OP("RestrictedTypeExample").Attr("t: {int32, float, bool}");
21 REGISTER_OP("NumberType").Attr("t: numbertype");
23 REGISTER_OP("EnumExample").Attr("e: {'apple', 'orange'}");
25 REGISTER_OP("MinIntExample").Attr("a: int >= 2");
27 REGISTER_OP("TypeListExample").Attr("a: list({int32, float}) >= 3");
29 REGISTER_OP("AttrDefaultExample").Attr("i: int = 0");
32 .Attr("s: string = 'foo'")
33 .Attr("i: int = 0")
34 .Attr("f: float = 1.0")
35 .Attr("
[all...]
/external/tensorflow/tensorflow/compiler/jit/ops/
H A Dxla_ops.cc22 .Attr("Tconstants: list(type) >= 0")
24 .Attr("Targs: list(type) >= 0")
26 .Attr("Nresources: int >= 0")
28 .Attr("Tresults: list(type) >= 0")
29 .Attr("function: func")
H A Dparallel_check_op.cc21 .Attr("T: list(type) >= 0")
/external/tensorflow/tensorflow/core/ops/
H A Dsendrecv_ops.cc23 .Attr("T: type")
24 .Attr("tensor_name: string")
25 .Attr("send_device: string")
26 .Attr("send_device_incarnation: int")
27 .Attr("recv_device: string")
28 .Attr("client_terminated: bool = false")
47 .Attr("tensor_type: type")
48 .Attr("tensor_name: string")
49 .Attr("send_device: string")
50 .Attr("send_device_incarnatio
[all...]
H A Dscript_ops.cc24 .Attr("token: string")
25 .Attr("Tin: list(type) >= 0")
26 .Attr("Tout: list(type) >=0")
33 .Attr("token: string")
34 .Attr("Tin: list(type) >= 0")
35 .Attr("Tout: list(type) >= 0")
41 .Attr("token: string")
42 .Attr("Tin: list(type) >= 0")
43 .Attr("Tout: list(type) >=0")
H A Dword2vec_ops.cc32 .Attr("filename: string")
33 .Attr("batch_size: int")
34 .Attr("window_size: int = 5")
35 .Attr("min_count: int = 5")
36 .Attr("subsample: float = 1e-3");
48 .Attr("vocab_count: list(int)")
49 .Attr("num_negative_samples: int");
H A Dmanip_ops.cc28 .Attr("T: type")
29 .Attr("Tshift: {int32,int64}")
30 .Attr("Taxis: {int32,int64}")
H A Dcandidate_sampling_ops.cc51 .Attr("num_true: int >= 1")
52 .Attr("num_sampled: int >= 1")
53 .Attr("unique: bool")
54 .Attr("range_max: int >= 1")
55 .Attr("seed: int = 0")
56 .Attr("seed2: int = 0")
65 .Attr("num_true: int >= 1")
66 .Attr("num_sampled: int >= 1")
67 .Attr("unique: bool")
68 .Attr("range_ma
[all...]
H A Dfunction_ops.cc23 .Attr("T: type")
24 .Attr("index: int >= 0")
39 .Attr("T: type")
40 .Attr("index: int >= 0")
55 .Attr("Tin: list(type)")
56 .Attr("T: type")
57 .Attr("N: int >= 1")
65 .Attr("T: type")
66 .Attr("N: int >= 1")
67 .Attr("out_type
[all...]
H A Drandom_ops.cc30 .Attr("seed: int = 0")
31 .Attr("seed2: int = 0")
32 .Attr("dtype: {half,bfloat16,float,double}")
33 .Attr("T: {int32, int64}")
42 .Attr("seed: int = 0")
43 .Attr("seed2: int = 0")
44 .Attr("Tout: {int32, int64}")
45 .Attr("T: {int32, int64}")
52 .Attr("seed: int = 0")
53 .Attr("seed
[all...]
H A Dfunctional_ops.cc27 .Attr("Tin: list(type)")
28 .Attr("Tout: list(type)")
29 .Attr("f: func")
47 .Attr("Tin: list(type)")
48 .Attr("Tout: list(type)")
49 .Attr("f: func")
56 .Attr("Tcond: type")
57 .Attr("Tin: list(type)")
58 .Attr("Tout: list(type)")
59 .Attr("then_branc
[all...]
H A Dlogging_ops.cc26 .Attr("T: list(type)")
27 .Attr("summarize: int = 3")
35 .Attr("T: type")
36 .Attr("U: list(type) >= 0")
37 .Attr("message: string = ''")
38 .Attr("first_n: int = -1")
39 .Attr("summarize: int = 3")
53 .Attr("T: type")
59 .Attr("T: type")
60 .Attr("descriptio
[all...]
H A Ddataset_ops.cc38 .Attr("Toutput_types: list(type) >= 1")
39 .Attr("output_shapes: list(shape) >= 1")
50 .Attr("Toutput_types: list(type) >= 1")
51 .Attr("output_shapes: list(shape) >= 1")
64 .Attr("Tvalues: type")
72 .Attr("output_types: list(type) >= 1")
73 .Attr("output_shapes: list(shape) >= 1")
74 .Attr("N: int >= 1")
81 .Attr("output_types: list(type) >= 1")
82 .Attr("output_shape
[all...]
H A Dstate_ops.cc26 .Attr("shape: shape")
27 .Attr("dtype: type")
28 .Attr("container: string = ''")
29 .Attr("shared_name: string = ''")
35 .Attr("shape: shape")
36 .Attr("dtype: type")
37 .Attr("container: string = ''")
38 .Attr("shared_name: string = ''")
60 .Attr("dtype: type")
66 .Attr("shap
[all...]
H A Dbatch_ops.cc27 .Attr("num_batch_threads: int")
28 .Attr("max_batch_size: int")
29 .Attr("max_enqueued_batches: int = 10")
30 .Attr("batch_timeout_micros: int")
31 .Attr("allowed_batch_sizes: list(int) = []")
32 .Attr("grad_timeout_micros: int")
33 .Attr("container: string = ''")
34 .Attr("shared_name: string = ''")
35 .Attr("batching_queue: string = ''")
36 .Attr("
[all...]
/external/tensorflow/tensorflow/contrib/cloud/ops/
H A Dbigquery_reader_ops.cc25 .Attr("container: string = ''")
26 .Attr("shared_name: string = ''")
27 .Attr("project_id: string")
28 .Attr("dataset_id: string")
29 .Attr("table_id: string")
30 .Attr("columns: list(string)")
31 .Attr("timestamp_millis: int")
32 .Attr("test_end_point: string = ''")
58 .Attr("project_id: string")
59 .Attr("dataset_i
[all...]
/external/tensorflow/tensorflow/compiler/tf2xla/ops/
H A Dfunctional_ops.cc26 .Attr("T: list(type) >= 0")
27 .Attr("cond: func")
28 .Attr("body: func")
53 .Attr("Tcond: type")
54 .Attr("then_branch: func")
55 .Attr("else_branch: func")
56 .Attr("Tin: list(type) >= 0")
57 .Attr("Tout: list(type) >= 0")
/external/tensorflow/tensorflow/contrib/tensorrt/ops/
H A Dtrt_engine_op.cc31 .Attr("serialized_engine: string")
32 .Attr("input_nodes: list(string)")
33 .Attr("output_nodes: list(string)")
34 .Attr("InT: list({float32})")
35 .Attr("OutT: list({float32})")
/external/tensorflow/tensorflow/user_ops/
H A Dinvalid_op.cc22 .Attr("invalid attr: int32") // invalid since the name has a space.
/external/tensorflow/tensorflow/cc/framework/
H A Dtest_op.cc25 .Attr("scope: int")
26 .Attr("builder: int = 1")
27 .Attr("while: int")
38 .Attr("scope: int = 2")
39 .Attr("throw_away2: int = 2")
40 .Attr("attrs: int = 4")
41 .Attr("node: int = 4")
54 .Attr("node: int = 4")
/external/tensorflow/tensorflow/contrib/data/ops/
H A Ddataset_ops.cc23 .Attr("output_types: list(type) >= 1")
24 .Attr("output_shapes: list(shape) >= 1")
34 .Attr("shared_name: string")
35 .Attr("container: string")
36 .Attr("f: func")
37 .Attr("buffer_size: int")
38 .Attr("thread_pool_size: int")
57 .Attr("output_types: list(type)")
/external/tensorflow/tensorflow/contrib/nccl/ops/
H A Dnccl_ops.cc27 .Attr("reduction: {'min', 'max', 'prod', 'sum'}")
28 .Attr("T: {float, float64, int32, int64}")
29 .Attr("num_devices: int")
30 .Attr("shared_name: string")
53 .Attr("reduction: {'min', 'max', 'prod', 'sum'}")
54 .Attr("T: {float, float64, int32, int64}")
55 .Attr("num_devices: int")
71 .Attr("reduction: {'min', 'max', 'prod', 'sum'}")
72 .Attr("T: {float, float64, int32, int64}")
73 .Attr("num_device
[all...]
/external/tensorflow/tensorflow/contrib/image/ops/
H A Dsingle_image_random_dot_stereograms_ops.cc27 .Attr("T: {double,float,int64,int32}")
30 .Attr("hidden_surface_removal: bool = true")
31 .Attr("convergence_dots_size: int = 8")
32 .Attr("dots_per_inch: int = 72")
33 .Attr("eye_separation: float = 2.5")
34 .Attr("mu: float = .3333")
35 .Attr("normalize: bool = true")
36 .Attr("normalize_max: float = -100.0")
37 .Attr("normalize_min: float = 100.0")
38 .Attr("border_leve
[all...]
/external/tensorflow/tensorflow/contrib/periodic_resample/ops/
H A Darray_ops.cc25 .Attr("T: numbertype")
27 .Attr("shape: shape")
/external/tensorflow/tensorflow/contrib/tpu/ops/
H A Dreplication_ops.cc26 .Attr("num_replicas: int >= 0")
27 .Attr("topology: string = \"\"")
28 .Attr("device_assignment: list(int) = []")
29 .Attr("computation_shape: list(int) = []")
35 .Attr("N: int >= 1")
36 .Attr("T: type")
54 .Attr("num_replicas: int >= 1")
55 .Attr("T: type")
67 .Attr("computation: func")
68 .Attr("num_replica
[all...]

Completed in 2291 milliseconds

1234567891011>>