Searched refs:TypeConstraint (Results 1 – 25 of 407) sorted by relevance
12345678910>>...17
| /external/tensorflow/tensorflow/core/kernels/ |
| D | shape_ops.cc | 29 .TypeConstraint<int32>("out_type"),
34 .TypeConstraint<int64_t>("out_type"),
42 .TypeConstraint<int32>("out_type") \
43 .TypeConstraint<type>("T"), \
48 .TypeConstraint<int64_t>("out_type") \
49 .TypeConstraint<type>("T"), \
64 .TypeConstraint<int32>("T")
65 .TypeConstraint<int32>("out_type"),
71 .TypeConstraint<int32>("T")
72 .TypeConstraint<int64_t>("out_type"),
[all …]
|
| D | segment_reduction_ops_impl_5.cc | 32 .TypeConstraint<type>("T") \
33 .TypeConstraint<index_type>("Tidx") \
34 .TypeConstraint<segment_ids_type>("Tsegmentids"), \
40 .TypeConstraint<type>("T") \
41 .TypeConstraint<index_type>("Tidx") \
42 .TypeConstraint<segment_ids_type>("Tsegmentids"), \
52 .TypeConstraint<type>("T") \
53 .TypeConstraint<index_type>("Tidx") \
54 .TypeConstraint<segment_ids_type>("Tsegmentids"), \
60 .TypeConstraint<type>("T") \
[all …]
|
| D | reshape_op.cc | 24 .TypeConstraint<int32>("Tshape"),
29 .TypeConstraint<int64_t>("Tshape"),
36 .TypeConstraint<type>("T") \
37 .TypeConstraint<int32>("Tshape"), \
42 .TypeConstraint<type>("T") \
43 .TypeConstraint<int64_t>("Tshape"), \
59 .TypeConstraint<int32>("T")
60 .TypeConstraint<int32>("Tshape"),
67 .TypeConstraint<int32>("T")
68 .TypeConstraint<int64_t>("Tshape"),
[all …]
|
| D | identity_op.cc | 58 Name("Identity").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
61 Name("PreventGradient").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
64 Name("RefIdentity").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
67 Name("StopGradient").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
71 .TypeConstraint<type>("T"), \
75 .TypeConstraint<type>("dtype"), \
78 Name("_EagerConst").Device(DEVICE_GPU).TypeConstraint<type>("T"), \
89 Name("Identity").Device(DEVICE_DEFAULT).TypeConstraint<type>("T"), \
93 .TypeConstraint<type>("T"), \
96 Name("RefIdentity").Device(DEVICE_DEFAULT).TypeConstraint<type>("T"), \
[all …]
|
| D | reduction_ops_sum.cc | 24 .TypeConstraint<type>("T") \
25 .TypeConstraint<int32>("Tidx"), \
30 .TypeConstraint<type>("T") \
31 .TypeConstraint<int64_t>("Tidx"), \
42 .TypeConstraint<type>("T") \
43 .TypeConstraint<int32>("Tidx") \
49 .TypeConstraint<type>("T") \
50 .TypeConstraint<int64_t>("Tidx") \
66 .TypeConstraint<int32>("T")
67 .TypeConstraint<int32>("Tidx")
[all …]
|
| D | argmax_op.cc | 133 .TypeConstraint<type>("T") \
134 .TypeConstraint<int64_t>("output_type") \
139 .TypeConstraint<type>("T") \
140 .TypeConstraint<int64_t>("output_type") \
145 .TypeConstraint<type>("T") \
146 .TypeConstraint<int32>("output_type") \
151 .TypeConstraint<type>("T") \
152 .TypeConstraint<int32>("output_type") \
157 .TypeConstraint<type>("T") \
158 .TypeConstraint<int16>("output_type") \
[all …]
|
| D | reduction_ops_min.cc | 24 .TypeConstraint<type>("T") \
25 .TypeConstraint<int32>("Tidx"), \
31 .TypeConstraint<type>("T") \
32 .TypeConstraint<int64_t>("Tidx"), \
44 .TypeConstraint<type>("T") \
45 .TypeConstraint<int32>("Tidx") \
52 .TypeConstraint<type>("T") \
53 .TypeConstraint<int64_t>("Tidx") \
73 .TypeConstraint<int32>("T")
74 .TypeConstraint<int32>("Tidx"),
[all …]
|
| D | reduction_ops_max.cc | 24 .TypeConstraint<type>("T") \
25 .TypeConstraint<int32>("Tidx"), \
31 .TypeConstraint<type>("T") \
32 .TypeConstraint<int64_t>("Tidx"), \
44 .TypeConstraint<type>("T") \
45 .TypeConstraint<int32>("Tidx") \
52 .TypeConstraint<type>("T") \
53 .TypeConstraint<int64_t>("Tidx") \
75 .TypeConstraint<int32>("T")
76 .TypeConstraint<int32>("Tidx"),
[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 | reduction_ops_prod.cc | 23 .TypeConstraint<type>("T") \
24 .TypeConstraint<int32>("Tidx"), \
29 .TypeConstraint<type>("T") \
30 .TypeConstraint<int64_t>("Tidx"), \
41 .TypeConstraint<type>("T") \
42 .TypeConstraint<int32>("Tidx") \
48 .TypeConstraint<type>("T") \
49 .TypeConstraint<int64_t>("Tidx") \
55 .TypeConstraint<type>("T") \
56 .TypeConstraint<uint64>("Tidx") \
|
| D | relu_op.cc | 36 Name("Relu").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
39 Name("ReluGrad").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
42 Name("Relu6").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
45 Name("Relu6Grad").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
48 Name("LeakyReluGrad").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
58 Name("LeakyRelu").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
69 Name("Elu").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
72 Name("EluGrad").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
75 Name("Selu").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
78 Name("SeluGrad").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
[all …]
|
| D | scan_ops.cc | 125 .TypeConstraint<type>("T") \
126 .TypeConstraint<int32>("Tidx"), \
131 .TypeConstraint<type>("T") \
132 .TypeConstraint<int64_t>("Tidx"), \
142 .TypeConstraint<type>("T") \
143 .TypeConstraint<int32>("Tidx") \
149 .TypeConstraint<type>("T") \
150 .TypeConstraint<int64_t>("Tidx") \
164 .TypeConstraint<type>("T") \
165 .TypeConstraint<int32>("Tidx"), \
[all …]
|
| D | unique_op.cc | 245 .TypeConstraint<type>("T") \
246 .TypeConstraint<int32>("out_idx"), \
250 .TypeConstraint<type>("T") \
251 .TypeConstraint<int64_t>("out_idx"), \
255 .TypeConstraint<type>("T") \
256 .TypeConstraint<int32>("out_idx"), \
260 .TypeConstraint<type>("T") \
261 .TypeConstraint<int64_t>("out_idx"), \
265 .TypeConstraint<type>("T") \
266 .TypeConstraint<int32>("out_idx"), \
[all …]
|
| D | dequantize_op.cc | 249 .TypeConstraint<quint8>("T")
250 .TypeConstraint<float>("dtype"),
254 .TypeConstraint<qint8>("T")
255 .TypeConstraint<float>("dtype"),
259 .TypeConstraint<quint16>("T")
260 .TypeConstraint<float>("dtype"),
264 .TypeConstraint<qint16>("T")
265 .TypeConstraint<float>("dtype"),
269 .TypeConstraint<qint32>("T")
270 .TypeConstraint<float>("dtype"),
[all …]
|
| D | reduction_ops_mean.cc | 24 .TypeConstraint<type>("T") \
25 .TypeConstraint<int32>("Tidx"), \
30 .TypeConstraint<type>("T") \
31 .TypeConstraint<int64_t>("Tidx"), \
42 .TypeConstraint<type>("T") \
43 .TypeConstraint<int32>("Tidx") \
49 .TypeConstraint<type>("T") \
50 .TypeConstraint<int64_t>("Tidx") \
|
| D | reduction_ops_euclidean.cc | 23 .TypeConstraint<type>("T") \
24 .TypeConstraint<int32>("Tidx"), \
29 .TypeConstraint<type>("T") \
30 .TypeConstraint<int64_t>("Tidx"), \
41 .TypeConstraint<type>("T") \
42 .TypeConstraint<int32>("Tidx") \
48 .TypeConstraint<type>("T") \
49 .TypeConstraint<int64_t>("Tidx") \
|
| D | control_flow_ops.cc | 68 .TypeConstraint<type>("T"), \
73 .TypeConstraint<type>("T"), \
80 .TypeConstraint<type>("T"), \
87 .TypeConstraint<type>("T"), \
92 .TypeConstraint<type>("T"), \
99 .TypeConstraint<type>("T"), \
130 .TypeConstraint<type>("T"), \
137 .TypeConstraint<type>("T"), \
147 .TypeConstraint<type>("T"), \
195 .TypeConstraint<type>("T"), \
[all …]
|
| D | pad_op.cc | 260 .TypeConstraint<type>("T") \
261 .TypeConstraint<int32>("Tpaddings") \
266 .TypeConstraint<type>("T") \
267 .TypeConstraint<int64_t>("Tpaddings") \
272 .TypeConstraint<type>("T") \
273 .TypeConstraint<int32>("Tpaddings") \
279 .TypeConstraint<type>("T") \
280 .TypeConstraint<int64_t>("Tpaddings") \
326 .TypeConstraint<T>("T") \
327 .TypeConstraint<int32>("Tpaddings") \
[all …]
|
| D | debug_ops.cc | 52 Name("DebugNanCount").Device(DEVICE_CPU).TypeConstraint<type>("T"), \
61 .TypeConstraint<type>("T"), \
71 .TypeConstraint<type>("T"), \
83 .TypeConstraint<type>("T"), \
104 .TypeConstraint<type>("T") \
105 .TypeConstraint<float>("output_dtype"), \
118 .TypeConstraint<type>("T") \
119 .TypeConstraint<double>("output_dtype"), \
134 .TypeConstraint<in_type>("T") \
135 .TypeConstraint<out_type>("output_dtype"), \
|
| D | roll_op.cc | 329 .TypeConstraint<type>("T") \
330 .TypeConstraint<int32>("Tshift") \
331 .TypeConstraint<int32>("Taxis") \
337 .TypeConstraint<type>("T") \
338 .TypeConstraint<int64_t>("Tshift") \
339 .TypeConstraint<int32>("Taxis") \
345 .TypeConstraint<type>("T") \
346 .TypeConstraint<int32>("Tshift") \
347 .TypeConstraint<int64_t>("Taxis") \
353 .TypeConstraint<type>("T") \
[all …]
|
| D | stack_ops.cc | 66 .TypeConstraint<type>("T"), \
71 .TypeConstraint<type>("T"), \
76 .TypeConstraint<type>("elem_type"), \
81 .TypeConstraint<type>("elem_type"), \
96 .TypeConstraint<type>("T"), \
103 .TypeConstraint<type>("T"), \
109 .TypeConstraint<type>("elem_type"), \
115 .TypeConstraint<type>("elem_type"), \
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_embedding_enqueue_ops.cc | 203 .TypeConstraint<int32>("T1")
204 .TypeConstraint<int32>("T2")
205 .TypeConstraint<float>("T3")
210 .TypeConstraint<int64>("T1")
211 .TypeConstraint<int64>("T2")
212 .TypeConstraint<float>("T3")
217 .TypeConstraint<int32>("T1")
218 .TypeConstraint<int64>("T2")
219 .TypeConstraint<float>("T3")
224 .TypeConstraint<int64>("T1")
[all …]
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device_ops.h | 79 Name("Const").Device(DEVICE).TypeConstraint("dtype", TYPES), \
82 Name("Identity").Device(DEVICE).TypeConstraint("T", TYPES), IdentityOp); \
101 .TypeConstraint<int32>("out_type") \
102 .TypeConstraint("T", TYPES), \
107 .TypeConstraint<int64_t>("out_type") \
108 .TypeConstraint("T", TYPES), \
113 .TypeConstraint<int32>("out_type") \
114 .TypeConstraint("T", TYPES), \
119 .TypeConstraint<int64_t>("out_type") \
120 .TypeConstraint("T", TYPES), \
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_tmp_bf16_ops.cc | 45 Name("Conv2D").Device(DEVICE_CPU).TypeConstraint<T>("T"), \
48 Name("Conv2DBackpropFilter").Device(DEVICE_CPU).TypeConstraint<T>("T"), \
51 Name("Conv2DBackpropInput").Device(DEVICE_CPU).TypeConstraint<T>("T"), \
54 Name("_FusedConv2D").Device(DEVICE_CPU).TypeConstraint<T>("T"), \
57 Name("AvgPool").Device(DEVICE_CPU).TypeConstraint<T>("T"), \
60 Name("AvgPoolGrad").Device(DEVICE_CPU).TypeConstraint<T>("T"), \
64 .TypeConstraint<bfloat16>("T") \
65 .TypeConstraint<float>("U"), \
69 .TypeConstraint<bfloat16>("T") \
70 .TypeConstraint<float>("U"), \
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | fft_ops.cc | 118 REGISTER_XLA_OP(Name("FFT").TypeConstraint("Tcomplex",
121 REGISTER_XLA_OP(Name("FFT2D").TypeConstraint("Tcomplex",
124 REGISTER_XLA_OP(Name("FFT3D").TypeConstraint("Tcomplex",
134 REGISTER_XLA_OP(Name("IFFT").TypeConstraint("Tcomplex", DT_COMPLEX64),
136 REGISTER_XLA_OP(Name("IFFT2D").TypeConstraint("Tcomplex", DT_COMPLEX64),
138 REGISTER_XLA_OP(Name("IFFT3D").TypeConstraint("Tcomplex", DT_COMPLEX64),
148 .TypeConstraint("Treal", DT_FLOAT)
149 .TypeConstraint("Tcomplex", DT_COMPLEX64)
153 .TypeConstraint("Treal", DT_FLOAT)
154 .TypeConstraint("Tcomplex", DT_COMPLEX64)
[all …]
|
12345678910>>...17