Searched refs:TypeConstraint (Results 1 – 25 of 313) sorted by relevance
12345678910>>...13
| /external/tensorflow/tensorflow/core/kernels/ |
| D | shape_ops.cc | 28 .TypeConstraint<int32>("out_type"),
33 .TypeConstraint<int64>("out_type"),
41 .TypeConstraint<int32>("out_type") \
42 .TypeConstraint<type>("T"), \
47 .TypeConstraint<int64>("out_type") \
48 .TypeConstraint<type>("T"), \
59 .TypeConstraint<int32>("T")
60 .TypeConstraint<int32>("out_type"),
66 .TypeConstraint<int32>("T")
67 .TypeConstraint<int64>("out_type"),
[all …]
|
| D | reshape_op.cc | 24 .TypeConstraint<int32>("Tshape"),
29 .TypeConstraint<int64>("Tshape"),
36 .TypeConstraint<type>("T") \
37 .TypeConstraint<int32>("Tshape"), \
42 .TypeConstraint<type>("T") \
43 .TypeConstraint<int64>("Tshape"), \
54 .TypeConstraint<type>("T") \
55 .TypeConstraint<int32>("Tshape"), \
60 .TypeConstraint<type>("T") \
61 .TypeConstraint<int64>("Tshape"), \
[all …]
|
| D | reduction_ops_sum.cc | 24 .TypeConstraint<type>("T") \
25 .TypeConstraint<int32>("Tidx"), \
30 .TypeConstraint<type>("T") \
31 .TypeConstraint<int64>("Tidx"), \
42 .TypeConstraint<type>("T") \
43 .TypeConstraint<int32>("Tidx") \
49 .TypeConstraint<type>("T") \
50 .TypeConstraint<int64>("Tidx") \
65 .TypeConstraint<int32>("T")
66 .TypeConstraint<int32>("Tidx")
[all …]
|
| D | reduction_ops_max.cc | 24 .TypeConstraint<type>("T") \
25 .TypeConstraint<int32>("Tidx"), \
30 .TypeConstraint<type>("T") \
31 .TypeConstraint<int64>("Tidx"), \
42 .TypeConstraint<type>("T") \
43 .TypeConstraint<int32>("Tidx") \
49 .TypeConstraint<type>("T") \
50 .TypeConstraint<int64>("Tidx") \
68 .TypeConstraint<int32>("T")
69 .TypeConstraint<int32>("Tidx"),
[all …]
|
| D | reduction_ops_min.cc | 24 .TypeConstraint<type>("T") \
25 .TypeConstraint<int32>("Tidx"), \
30 .TypeConstraint<type>("T") \
31 .TypeConstraint<int64>("Tidx"), \
42 .TypeConstraint<type>("T") \
43 .TypeConstraint<int32>("Tidx") \
49 .TypeConstraint<type>("T") \
50 .TypeConstraint<int64>("Tidx") \
66 .TypeConstraint<int32>("T")
67 .TypeConstraint<int32>("Tidx"),
[all …]
|
| D | relu_op.cc | 38 Name("Relu").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
41 Name("ReluGrad").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
44 Name("Relu6").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
47 Name("Relu6Grad").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
50 Name("LeakyRelu").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
53 Name("LeakyReluGrad").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
61 Name("Elu").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
64 Name("EluGrad").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
67 Name("Selu").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
70 Name("SeluGrad").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
[all …]
|
| D | unique_op.cc | 200 .TypeConstraint<type>("T") \
201 .TypeConstraint<int32>("out_idx"), \
205 .TypeConstraint<type>("T") \
206 .TypeConstraint<int64>("out_idx"), \
210 .TypeConstraint<type>("T") \
211 .TypeConstraint<int32>("out_idx"), \
215 .TypeConstraint<type>("T") \
216 .TypeConstraint<int64>("out_idx"), \
220 .TypeConstraint<type>("T") \
221 .TypeConstraint<int32>("out_idx"), \
[all …]
|
| D | pad_op.cc | 268 .TypeConstraint<type>("T") \
269 .TypeConstraint<int32>("Tpaddings") \
274 .TypeConstraint<type>("T") \
275 .TypeConstraint<int64>("Tpaddings") \
280 .TypeConstraint<type>("T") \
281 .TypeConstraint<int32>("Tpaddings") \
287 .TypeConstraint<type>("T") \
288 .TypeConstraint<int64>("Tpaddings") \
332 .TypeConstraint<T>("T") \
333 .TypeConstraint<int32>("Tpaddings") \
[all …]
|
| D | identity_op.cc | 51 Name("Identity").Device(DEVICE_SYCL).TypeConstraint<type>("T"), \
54 Name("PreventGradient").Device(DEVICE_SYCL).TypeConstraint<type>("T"), \
57 Name("RefIdentity").Device(DEVICE_SYCL).TypeConstraint<type>("T"), \
60 Name("StopGradient").Device(DEVICE_SYCL).TypeConstraint<type>("T"), \
72 .TypeConstraint<type>("T"), \
78 .TypeConstraint<type>("T"), \
90 Name("Identity").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
93 Name("PreventGradient").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
96 Name("RefIdentity").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
99 Name("StopGradient").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
[all …]
|
| D | control_flow_ops.cc | 46 .TypeConstraint<type>("T"), \
53 .TypeConstraint<type>("T"), \
60 .TypeConstraint<type>("T"), \
67 .TypeConstraint<type>("T"), \
98 .TypeConstraint<type>("T"), \
108 .TypeConstraint<type>("T"), \
127 .TypeConstraint<type>("T"), \
135 .TypeConstraint<type>("T"), \
149 .TypeConstraint<type>("T"), \
163 .TypeConstraint<type>("T"), \
[all …]
|
| D | reduction_ops_euclidean.cc | 23 .TypeConstraint<type>("T") \
24 .TypeConstraint<int32>("Tidx"), \
29 .TypeConstraint<type>("T") \
30 .TypeConstraint<int64>("Tidx"), \
41 .TypeConstraint<type>("T") \
42 .TypeConstraint<int32>("Tidx") \
48 .TypeConstraint<type>("T") \
49 .TypeConstraint<int64>("Tidx") \
64 .TypeConstraint<type>("T") \
65 .TypeConstraint<int32>("Tidx") \
[all …]
|
| D | reduction_ops_mean.cc | 24 .TypeConstraint<type>("T") \
25 .TypeConstraint<int32>("Tidx"), \
30 .TypeConstraint<type>("T") \
31 .TypeConstraint<int64>("Tidx"), \
42 .TypeConstraint<type>("T") \
43 .TypeConstraint<int32>("Tidx") \
49 .TypeConstraint<type>("T") \
50 .TypeConstraint<int64>("Tidx") \
65 .TypeConstraint<type>("T") \
66 .TypeConstraint<int32>("Tidx") \
[all …]
|
| D | reduction_ops_prod.cc | 23 .TypeConstraint<type>("T") \
24 .TypeConstraint<int32>("Tidx"), \
29 .TypeConstraint<type>("T") \
30 .TypeConstraint<int64>("Tidx"), \
41 .TypeConstraint<type>("T") \
42 .TypeConstraint<int32>("Tidx") \
48 .TypeConstraint<type>("T") \
49 .TypeConstraint<int64>("Tidx") \
65 .TypeConstraint<type>("T") \
66 .TypeConstraint<int32>("Tidx") \
[all …]
|
| D | argmax_op.cc | 128 .TypeConstraint<type>("T") \
129 .TypeConstraint<int64>("output_type") \
134 .TypeConstraint<type>("T") \
135 .TypeConstraint<int64>("output_type") \
140 .TypeConstraint<type>("T") \
141 .TypeConstraint<int32>("output_type") \
146 .TypeConstraint<type>("T") \
147 .TypeConstraint<int32>("output_type") \
198 .TypeConstraint<type>("T") \
199 .TypeConstraint<int64>("output_type") \
[all …]
|
| D | list_kernels.cu.cc | 41 .TypeConstraint<T>("element_dtype") \
46 .TypeConstraint<T>("element_dtype") \
52 .TypeConstraint<T>("element_dtype") \
58 .TypeConstraint<T>("element_dtype") \
63 .TypeConstraint<T>("element_dtype") \
68 .TypeConstraint<T>("element_dtype") \
75 .TypeConstraint<T>("element_dtype") \
79 .TypeConstraint<T>("element_dtype") \
84 .TypeConstraint<T>("element_dtype") \
90 .TypeConstraint<T>("element_dtype") \
[all …]
|
| D | scan_ops.cc | 121 .TypeConstraint<type>("T") \
122 .TypeConstraint<int32>("Tidx"), \
127 .TypeConstraint<type>("T") \
128 .TypeConstraint<int64>("Tidx"), \
138 .TypeConstraint<type>("T") \
139 .TypeConstraint<int32>("Tidx") \
145 .TypeConstraint<type>("T") \
146 .TypeConstraint<int64>("Tidx") \
158 .TypeConstraint<type>("T") \
159 .TypeConstraint<int32>("Tidx"), \
[all …]
|
| D | mkl_conv_ops.cc | 842 .TypeConstraint<float>("T")
847 .TypeConstraint<float>("T")
1849 .TypeConstraint<quint8>("Tinput")
1850 .TypeConstraint<qint8>("Tfilter")
1851 .TypeConstraint<qint32>("out_type"),
1856 .TypeConstraint<quint8>("Tinput")
1857 .TypeConstraint<qint8>("Tfilter")
1858 .TypeConstraint<qint8>("out_type"),
1865 .TypeConstraint<quint8>("Tinput")
1866 .TypeConstraint<qint8>("Tfilter")
[all …]
|
| D | stack_ops.cc | 71 .TypeConstraint<type>("T"), \
76 .TypeConstraint<type>("T"), \
91 .TypeConstraint<type>("T"), \
98 .TypeConstraint<type>("T"), \
111 .TypeConstraint<type>("T"), \
122 .TypeConstraint<type>("T"), \
138 .TypeConstraint<type>("elem_type"), \
143 .TypeConstraint<type>("elem_type"), \
157 .TypeConstraint<type>("elem_type"), \
163 .TypeConstraint<type>("elem_type"), \
[all …]
|
| D | searchsorted_op.cc | 165 .TypeConstraint<type>("T") \
166 .TypeConstraint<int32>("out_type"), \
175 .TypeConstraint<type>("T") \
176 .TypeConstraint<int64>("out_type"), \
187 .TypeConstraint<type>("T") \
188 .TypeConstraint<int32>("out_type"), \
197 .TypeConstraint<type>("T") \
198 .TypeConstraint<int64>("out_type"), \
209 .TypeConstraint<type>("T") \
210 .TypeConstraint<int32>("out_type"), \
[all …]
|
| D | reverse_op.cc | 301 .TypeConstraint<T>("T") \
306 .TypeConstraint<T>("T") \
307 .TypeConstraint<int32>("Tidx") \
312 .TypeConstraint<T>("T") \
313 .TypeConstraint<int64>("Tidx") \
359 .TypeConstraint<T>("T") \
364 .TypeConstraint<T>("T") \
365 .TypeConstraint<int32>("Tidx") \
370 .TypeConstraint<T>("T") \
371 .TypeConstraint<int64>("Tidx") \
[all …]
|
| D | function_ops.cc | 79 Name(kArgOp).Device(DEVICE_SYCL).TypeConstraint<type>("T"), ArgOp);
84 .TypeConstraint<int32>("T"),
89 Name(kRetOp).Device(DEVICE_SYCL).TypeConstraint<type>("T"), RetvalOp);
94 .TypeConstraint<int32>("T"),
101 Name(kArgOp).Device(DEVICE_GPU).TypeConstraint<type>("T"), ArgOp);
107 .TypeConstraint<int32>("T"),
110 Name(kDeviceArgOp).Device(DEVICE_GPU).TypeConstraint<int32>("T"), ArgOp);
116 .TypeConstraint<ResourceHandle>("T"),
122 .TypeConstraint<string>("T"),
126 Name(kArgOp).Device(DEVICE_GPU).TypeConstraint<Variant>("T"), ArgOp);
[all …]
|
| D | bcast_ops.cc | 121 .TypeConstraint<int32>("T")
128 .TypeConstraint<int64>("T")
135 .TypeConstraint<int32>("T")
142 .TypeConstraint<int64>("T")
151 .TypeConstraint<int32>("T")
158 .TypeConstraint<int64>("T")
167 .TypeConstraint<int32>("T")
175 .TypeConstraint<int64>("T")
183 .TypeConstraint<int32>("T")
191 .TypeConstraint<int64>("T")
[all …]
|
| D | tensor_array_ops.cc | 242 .TypeConstraint<type>("dtype") \
248 .TypeConstraint<type>("dtype") \
254 .TypeConstraint<type>("dtype") \
453 Name("TensorArrayWrite").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
456 Name("TensorArrayWriteV2").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
459 Name("TensorArrayWriteV3").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
471 .TypeConstraint<type>("T") \
477 .TypeConstraint<type>("T") \
483 .TypeConstraint<type>("T") \
540 .TypeConstraint<type>("dtype"), \
[all …]
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device_ops.h | 95 Name("Const").Device(DEVICE).TypeConstraint("dtype", TYPES), \
100 Name("Identity").Device(DEVICE).TypeConstraint("T", TYPES), IdentityOp); \
102 Name("Identity").Device(DEVICE).TypeConstraint("T", DT_STRING), \
105 Name("Identity").Device(DEVICE).TypeConstraint<Variant>("T"), \
109 .TypeConstraint<ResourceHandle>("T") \
136 .TypeConstraint<int32>("out_type") \
137 .TypeConstraint("T", TYPES), \
142 .TypeConstraint<int64>("out_type") \
143 .TypeConstraint("T", TYPES), \
148 .TypeConstraint<int32>("out_type") \
[all …]
|
| /external/tensorflow/tensorflow/contrib/reduce_slice_ops/kernels/ |
| D | reduce_slice_ops.cc | 160 .TypeConstraint<type>("T") \
161 .TypeConstraint<index_type>("Tindices"), \
166 .TypeConstraint<type>("T") \
167 .TypeConstraint<index_type>("Tindices"), \
174 .TypeConstraint<type>("T") \
175 .TypeConstraint<index_type>("Tindices"), \
180 .TypeConstraint<type>("T") \
181 .TypeConstraint<index_type>("Tindices"), \
207 .TypeConstraint<type>("T") \
208 .TypeConstraint<index_type>("Tindices"), \
[all …]
|
12345678910>>...13