| /external/tensorflow/tensorflow/core/ops/ |
| D | nn_ops.cc | 72 .Attr("ksize: list(int) >= 4")
73 .Attr("strides: list(int) >= 4")
74 .Attr(GetPaddingAttrString())
75 .Attr(GetConvnetDataFormatAttrString())
76 .Attr("T: {half, bfloat16, float, double}")
83 .Attr("ksize: list(int) >= 4")
84 .Attr("strides: list(int) >= 4")
85 .Attr(GetPaddingAttrString())
86 .Attr(GetConvnetDataFormatAttrString())
87 .Attr("T: {half, bfloat16, float, double}")
[all …]
|
| D | experimental_dataset_ops.cc | 25 .Attr("output_types: list(type) >= 1")
26 .Attr("output_shapes: list(shape) >= 1")
33 .Attr("output_types: list(type) >= 1")
34 .Attr("output_shapes: list(shape) >= 1")
47 .Attr("Targuments: list(type) >= 0")
48 .Attr("num_elements_per_branch: int >= 1")
49 .Attr("branches: list(func) >= 1")
50 .Attr("other_arguments_lengths: list(int) >= 1")
51 .Attr("output_types: list(type) >= 1")
52 .Attr("output_shapes: list(shape) >= 1")
[all …]
|
| D | mkl_nn_ops.cc | 46 .Attr("T: {float}")
47 .Attr("num_args: int >= 0")
48 .Attr("strides: list(int)")
49 .Attr("is_filter_const: bool = false")
50 .Attr(GetPaddingAttrString())
51 .Attr(GetConvnetDataFormatAttrString())
52 .Attr("dilations: list(int) = [1, 1, 1, 1]")
53 .Attr("fused_ops: list(string) = []")
55 .Attr("epsilon: float = 0.0001")
72 .Attr("T: {float}")
[all …]
|
| D | dataset_ops.cc | 38 .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")
74 .Attr("init_func: func")
75 .Attr("next_func: func")
76 .Attr("finalize_func: func")
77 .Attr("Tinit_func_args: list(type) >= 0")
78 .Attr("Tnext_func_args: list(type) >= 0")
[all …]
|
| D | state_ops.cc | 26 .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("shape: shape")
[all …]
|
| D | data_flow_ops.cc | 52 .Attr("num_partitions: int")
53 .Attr("T: type")
146 .Attr("N : int >= 1")
147 .Attr("T : type")
154 .Attr("N : int >= 1")
155 .Attr("T : type")
182 .Attr("component_types: list(type) >= 1")
183 .Attr("shapes: list(shape) >= 0 = []")
184 .Attr("capacity: int = -1")
185 .Attr("min_after_dequeue: int = 0")
[all …]
|
| D | candidate_sampling_ops.cc | 51 .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_max: int >= 1")
[all …]
|
| D | functional_ops.cc | 27 .Attr("Tin: list(type)")
28 .Attr("Tout: list(type)")
29 .Attr("f: func")
59 .Attr("Tin: list(type)")
60 .Attr("Tout: list(type)")
61 .Attr("f: func")
70 .Attr("Tcond: type")
71 .Attr("Tin: list(type)")
72 .Attr("Tout: list(type)")
73 .Attr("then_branch: func")
[all …]
|
| D | cudnn_rnn_ops.cc | 44 .Attr("T: {float16, float32, float64}")
45 .Attr("S: {int32, int64}")
46 .Attr(kRNNModeAttrs)
47 .Attr(kRNNInputModeAttrs)
48 .Attr(kRNNDirectionAttrs)
49 .Attr("dropout: float = 0.0")
50 .Attr("seed: int = 0")
51 .Attr("seed2: int = 0")
74 .Attr("T: {float16, float32, float64}")
75 .Attr(kRNNModeAttrs)
[all …]
|
| D | random_ops.cc | 30 .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}")
57 .Attr("seed: int = 0")
58 .Attr("seed2: int = 0")
[all …]
|
| D | sendrecv_ops.cc | 23 .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_incarnation: int")
[all …]
|
| D | training_ops.cc | 75 .Attr("T: numbertype")
76 .Attr("use_locking: bool = false")
83 .Attr("T: numbertype")
84 .Attr("use_locking: bool = false")
109 .Attr("T: numbertype")
110 .Attr("use_locking: bool = false")
123 .Attr("T: numbertype")
124 .Attr("Tindices: {int32, int64}")
125 .Attr("use_locking: bool = false")
136 .Attr("T: numbertype")
[all …]
|
| D | batch_ops.cc | 26 .Attr("f: func")
27 .Attr("num_batch_threads: int")
28 .Attr("max_batch_size: int")
29 .Attr("batch_timeout_micros: int")
30 .Attr("max_enqueued_batches: int = 10")
31 .Attr("allowed_batch_sizes: list(int) = []")
32 .Attr("container: string = ''")
33 .Attr("shared_name: string = ''")
34 .Attr("batching_queue: string = ''")
35 .Attr("Tin: list(type)")
[all …]
|
| D | collective_ops.cc | 24 .Attr("T: {float, float16, float64, int32, int64}")
25 .Attr("group_size: int")
26 .Attr("group_key: int")
27 .Attr("instance_key: int")
28 .Attr("merge_op: {'Min', 'Max', 'Mul', 'Add'}")
29 .Attr("final_op: {'Id', 'Div'}")
30 .Attr("subdiv_offsets: list(int)")
31 .Attr("wait_for: list(int) = []")
38 .Attr("T: {float, float16, float64, int32, int64}")
39 .Attr("group_size: int")
[all …]
|
| D | lookup_ops.cc | 75 .Attr("Tin: type")
76 .Attr("Tout: type")
163 .Attr("Tin: type")
164 .Attr("Tout: type")
191 .Attr("Tin: type")
192 .Attr("Tout: type")
207 .Attr("Tin: type")
208 .Attr("Tout: type")
220 .Attr("Tin: type")
246 .Attr("Tkeys: type")
[all …]
|
| D | tpu_replication_ops.cc | 26 .Attr("num_replicas: int >= 0")
27 .Attr("num_cores_per_replica: int = 1")
28 .Attr("topology: string = \"\"")
29 .Attr("use_tpu: bool = true")
30 .Attr("device_assignment: list(int) = []")
32 .Attr("computation_shape: list(int) = []")
33 .Attr("host_compute_core: list(string) = []")
34 .Attr("padding_map: list(string) = []")
35 .Attr("step_marker_location: string = \"STEP_MARK_AT_ENTRY\"")
41 .Attr("N: int >= 1")
[all …]
|
| D | math_ops.cc | 30 .Attr("N: int >= 1")
31 .Attr("T: {numbertype, variant}")
109 .Attr("N: int >= 1")
110 .Attr("T: numbertype")
111 .Attr("shape: shape")
122 .Attr(
125 .Attr("adj_x: bool = false")
126 .Attr("adj_y: bool = false")
171 .Attr("SrcT: type")
172 .Attr("DstT: type")
[all …]
|
| D | nccl_ops.cc | 27 .Attr("reduction: {'min', 'max', 'prod', 'sum'}")
28 .Attr("T: {half, float, float64, int32, int64}")
29 .Attr("num_devices: int")
30 .Attr("shared_name: string")
39 .Attr("reduction: {'min', 'max', 'prod', 'sum'}")
40 .Attr("T: {half, float, float64, int32, int64}")
41 .Attr("num_devices: int")
47 .Attr("reduction: {'min', 'max', 'prod', 'sum'}")
48 .Attr("T: {half, float, float64, int32, int64}")
49 .Attr("num_devices: int")
[all …]
|
| D | string_ops.cc | 43 .Attr("replace_global: bool = true")
54 .Attr("pattern: string")
55 .Attr("rewrite: string")
57 .Attr("replace_global: bool = true")
73 .Attr("pattern: string")
80 .Attr("num_buckets: int >= 1")
86 .Attr("num_buckets: int >= 1")
87 .Attr("key: list(int)")
93 .Attr("num_buckets: int >= 1")
99 .Attr("keep_dims: bool = false")
[all …]
|
| D | logging_ops.cc | 29 .Attr("T: list(type)")
30 .Attr("summarize: int = 3")
40 .Attr("T: type")
41 .Attr("U: list(type) >= 0")
42 .Attr("message: string = ''")
43 .Attr("first_n: int = -1")
44 .Attr("summarize: int = 3")
52 .Attr("output_stream: string = 'stderr'")
75 .Attr("T: type")
81 .Attr("T: type")
[all …]
|
| /external/clang/lib/Sema/ |
| D | SemaDeclAttr.cpp | 175 static unsigned getNumAttributeArgs(const AttributeList &Attr) { in getNumAttributeArgs() argument
177 return Attr.getNumArgs() + Attr.hasParsedType(); in getNumAttributeArgs()
181 static bool checkAttributeNumArgsImpl(Sema &S, const AttributeList &Attr, in checkAttributeNumArgsImpl() argument
184 if (Comp(getNumAttributeArgs(Attr), Num)) { in checkAttributeNumArgsImpl()
185 S.Diag(Attr.getLoc(), Diag) << Attr.getName() << Num; in checkAttributeNumArgsImpl()
194 static bool checkAttributeNumArgs(Sema &S, const AttributeList &Attr, in checkAttributeNumArgs() argument
196 return checkAttributeNumArgsImpl(S, Attr, Num, in checkAttributeNumArgs()
203 static bool checkAttributeAtLeastNumArgs(Sema &S, const AttributeList &Attr, in checkAttributeAtLeastNumArgs() argument
205 return checkAttributeNumArgsImpl(S, Attr, Num, in checkAttributeAtLeastNumArgs()
212 static bool checkAttributeAtMostNumArgs(Sema &S, const AttributeList &Attr, in checkAttributeAtMostNumArgs() argument
[all …]
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | test_ops.cc | 81 .Attr("T: {float, double}")
244 REGISTER_OP("CopyOp").Input("a: T").Output("b: T").Attr("T: type").SetShapeFn(
356 .Attr("N: int >= 1")
357 .Attr("T: type")
362 .Attr("T: list(type) >= 1")
365 REGISTER_OP("Unary").Input("a: T").Output("b: T").Attr("T: type").SetShapeFn(
370 .Attr("default_float: float = 123.0")
378 .Attr("foo: int = 1")
382 .Attr("a: list(string)")
383 .Attr("b: string")
[all …]
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_def_builder_test.cc | 87 TEST_F(OpDefBuilderTest, Attr) { in TEST_F() argument
88 ExpectSuccess(b().Attr("a:string"), "attr: { name: 'a' type: 'string' }"); in TEST_F()
89 ExpectSuccess(b().Attr("A: int"), "attr: { name: 'A' type: 'int' }"); in TEST_F()
90 ExpectSuccess(b().Attr("a1 :float"), "attr: { name: 'a1' type: 'float' }"); in TEST_F()
91 ExpectSuccess(b().Attr("a_a : bool"), "attr: { name: 'a_a' type: 'bool' }"); in TEST_F()
92 ExpectSuccess(b().Attr("aB : type"), "attr: { name: 'aB' type: 'type' }"); in TEST_F()
93 ExpectSuccess(b().Attr("aB_3\t: shape"), in TEST_F()
95 ExpectSuccess(b().Attr("t: tensor"), "attr: { name: 't' type: 'tensor' }"); in TEST_F()
96 ExpectSuccess(b().Attr("XYZ\t:\tlist(type)"), in TEST_F()
98 ExpectSuccess(b().Attr("f: func"), "attr { name: 'f' type: 'func'}"); in TEST_F()
[all …]
|
| D | node_def_builder_test.cc | 210 .Attr("foo", 12) in TEST_F()
220 .Attr("T: type")); in TEST_F()
233 ExpectSuccess(Builder().Input(FakeInput(DT_BOOL)).Attr("T", DT_BOOL), in TEST_F()
239 ExpectFailure(Builder().Input(FakeInput(DT_BOOL)).Attr("T", DT_STRING), in TEST_F()
242 ExpectFailure(Builder().Attr("T", DT_STRING).Input(FakeInput(DT_BOOL)), in TEST_F()
245 ExpectFailure(Builder().Attr("T", 12).Input(FakeInput(DT_BOOL)), in TEST_F()
250 Op(OpDefBuilder("PolymorphicOut").Output("out: T").Attr("T: type")); in TEST_F()
252 ExpectSuccess(Builder().Attr("T", DT_INT32), {}, {DT_INT32}, R"proto( in TEST_F()
256 ExpectSuccess(Builder().Attr("T", DT_FLOAT), {}, {DT_FLOAT}, R"proto( in TEST_F()
261 ExpectSuccess(Builder().Attr("T", DT_FLOAT).Attr("T", DT_FLOAT), {}, in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ops/ |
| D | xla_ops.cc | 42 .Attr("T: numbertype")
43 .Attr("Tindices: {int32, int64}")
63 .Attr("lower: bool")
64 .Attr("max_iter: int")
65 .Attr("epsilon: float")
69 .Attr("T: numbertype")
98 .Attr("max_iter: int")
99 .Attr("epsilon: float")
100 .Attr("precision_config: string")
105 .Attr("T: numbertype")
[all …]
|