Home
last modified time | relevance | path

Searched refs:Attr (Results 1 – 25 of 926) sorted by relevance

12345678910>>...38

/external/tensorflow/tensorflow/core/ops/
Dtpu_embedding_load_retrieve_ops.cc34 .Attr("table_id: int = -1")
35 .Attr("table_name: string = \"\"")
36 .Attr("num_shards: int")
37 .Attr("shard_id: int")
38 .Attr("config: string = \"\"")
46 .Attr("table_id: int = -1")
47 .Attr("table_name: string = \"\"")
48 .Attr("num_shards: int")
49 .Attr("shard_id: int")
50 .Attr("config: string = \"\"")
[all …]
Dmkl_nn_ops.cc39 .Attr("T: {bfloat16, float}")
40 .Attr("strides: list(int) >= 5")
41 .Attr("is_filter_const: bool = false")
42 .Attr(GetPaddingAttrString())
43 .Attr(GetConvnet3dDataFormatAttrString())
44 .Attr("dilations: list(int) = [1, 1, 1, 1, 1]")
59 .Attr("T: {bfloat16, float}")
60 .Attr("strides: list(int) >= 5")
61 .Attr("dilations: list(int) = [1, 1, 1, 1, 1]")
62 .Attr("Tshape: {int32, int64} = DT_INT32")
[all …]
Dexperimental_dataset_ops.cc24 .Attr("output_types: list(type) >= 1")
25 .Attr("output_shapes: list(shape) >= 1")
37 .Attr("output_types: list(type) >= 1")
38 .Attr("output_shapes: list(shape) >= 1")
50 .Attr("output_types: list(type) >= 1")
51 .Attr("output_shapes: list(shape) >= 1")
64 .Attr("auto_shard_policy: int = 0")
65 .Attr("output_types: list(type) >= 1")
66 .Attr("output_shapes: list(shape) >= 1")
67 .Attr("num_replicas: int = 0")
[all …]
Dnn_ops.cc75 .Attr("ksize: list(int) >= 4")
76 .Attr("strides: list(int) >= 4")
77 .Attr(GetPaddingAttrString())
78 .Attr(GetConvnetDataFormatAttrString())
79 .Attr("T: {half, bfloat16, float, double}")
86 .Attr("ksize: list(int) >= 4")
87 .Attr("strides: list(int) >= 4")
88 .Attr(GetPaddingAttrString())
89 .Attr(GetConvnetDataFormatAttrString())
90 .Attr("T: {half, bfloat16, float, double}")
[all …]
Drisc_ops.cc51 .Attr("T: {bfloat16, half, float, double}")
58 .Attr("T: {bfloat16, half, float, double}")
68 .Attr("op_type: {'ADD', 'SUB', 'MUL', 'DIV', 'REM', 'MIN', 'POW'}")
69 .Attr("T: {bfloat16, half, float, double}")
76 .Attr("op_type: {'EQ', 'NE', 'GE', 'GT', 'LE', 'LT'}")
77 .Attr("T: {bfloat16, half, float, double}")
84 .Attr("SrcT: type")
85 .Attr("DstT: type")
93 .Attr("T: type")
94 .Attr("Tidx: {int32, int64} = DT_INT32")
[all …]
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("T: type")
48 .Attr("tensor_name: string")
49 .Attr("send_device: string")
50 .Attr("send_device_incarnation: int")
[all …]
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")
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 …]
Ddebug_ops.cc31 .Attr("T: type")
32 .Attr("tensor_name: string = ''")
33 .Attr("debug_ops_spec: list(string) = []")
40 .Attr("T: type")
41 .Attr("tensor_name: string = ''")
42 .Attr("debug_ops_spec: list(string) = []")
49 .Attr("T: type")
50 .Attr("device_name: string = ''")
51 .Attr("tensor_name: string = ''")
52 .Attr("debug_urls: list(string) = []")
[all …]
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("shape: shape")
[all …]
Dcollective_ops.cc25 .Attr("T: {float, float16, float64, int32, int64}")
26 .Attr("group_size: int")
27 .Attr("group_key: int")
28 .Attr("instance_key: int")
29 .Attr("merge_op: {'Min', 'Max', 'Mul', 'Add'}")
30 .Attr("final_op: {'Id', 'Div'}")
31 .Attr("subdiv_offsets: list(int)")
32 .Attr("wait_for: list(int) = []")
33 .Attr("communication_hint: string = 'auto'")
34 .Attr("timeout_seconds: float = 0")
[all …]
Ddata_flow_ops.cc52 .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 …]
Dcudnn_rnn_ops.cc44 .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")
52 .Attr("num_proj: int = 0")
75 .Attr("T: {float16, float32, float64}")
[all …]
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_max: int >= 1")
[all …]
Dfunctional_ops.cc27 .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 …]
Dtpu_replication_ops.cc26 .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\"")
36 .Attr("allow_soft_placement: bool = false")
[all …]
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}")
67 .Attr("seed: int = 0")
68 .Attr("seed2: int = 0")
[all …]
Dbatch_ops.cc26 .Attr("f: func")
27 .Attr("num_batch_threads: int")
41 .Attr("max_batch_size: int")
42 .Attr("batch_timeout_micros: int")
43 .Attr("max_enqueued_batches: int = 10")
44 .Attr("allowed_batch_sizes: list(int) = []")
45 .Attr("container: string = ''")
46 .Attr("shared_name: string = ''")
47 .Attr("batching_queue: string = ''")
48 .Attr("Tin: list(type)")
[all …]
Dtraining_ops.cc90 .Attr("T: numbertype")
91 .Attr("use_locking: bool = false")
98 .Attr("T: numbertype")
99 .Attr("use_locking: bool = false")
124 .Attr("T: numbertype")
125 .Attr("use_locking: bool = false")
137 .Attr("T: numbertype")
138 .Attr("Tindices: {int32, int64}")
139 .Attr("use_locking: bool = false")
149 .Attr("T: numbertype")
[all …]
Dtpu_embedding_ops.cc63 .Attr("num_outputs: int >= 1")
64 .Attr("config: string")
91 .Attr("table_id: int >= 0")
92 .Attr("lookup_id: int >= 0")
101 .Attr("N: int >= 1")
102 .Attr("NN: int >= 0 = 0")
103 .Attr("config: string")
123 .Attr("N: int >= 1")
124 .Attr("device_ordinal: int = -1")
133 .Attr("T1: {int32,int64} = DT_INT32")
[all …]
Dmath_ops.cc30 .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")
133 .Attr(
136 .Attr("adj_x: bool = false")
[all …]
Dstring_ops.cc43 .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 …]
/external/clang/lib/Sema/
DSemaDeclAttr.cpp175 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/core/framework/
Dop_def_builder_test.cc87 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 …]
/external/tensorflow/tensorflow/compiler/mlir/tfr/resources/
Dtest_ops.cc26 .Attr("T: numbertype");
31 .Attr("N: int >= 1")
32 .Attr("T: numbertype");
37 .Attr("N: int >= 1")
38 .Attr("T: numbertype");
43 .Attr("N: int >= 1")
44 .Attr("T: numbertype");
50 .Attr("T: numbertype")
51 .Attr("pred: bool = false");
57 .Attr("N: int >= 1")
[all …]
/external/tensorflow/tensorflow/python/framework/
Dtest_ops.cc87 .Attr("T: {float, double}")
250 REGISTER_OP("CopyOp").Input("a: T").Output("b: T").Attr("T: type").SetShapeFn(
362 .Attr("N: int >= 1")
363 .Attr("T: type")
368 .Attr("T: list(type) >= 1")
371 REGISTER_OP("Unary").Input("a: T").Output("b: T").Attr("T: type").SetShapeFn(
376 .Attr("default_float: float = 123.0")
384 .Attr("foo: int = 1")
388 .Attr("a: list(string)")
389 .Attr("b: string")
[all …]

12345678910>>...38