Searched refs:Targmax (Results 1 – 16 of 16) sorted by relevance
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | MaxPoolWithArgmax.pbtxt | 13 type_attr: "Targmax"
28 name: "Targmax"
76 type_attr: "Targmax"
91 name: "Targmax"
143 type_attr: "Targmax"
158 name: "Targmax"
212 type_attr: "Targmax"
227 name: "Targmax"
282 type_attr: "Targmax"
297 name: "Targmax"
[all …]
|
| D | MaxPoolGradWithArgmax.pbtxt | 13 type_attr: "Targmax"
42 name: "Targmax"
77 type_attr: "Targmax"
106 name: "Targmax"
145 type_attr: "Targmax"
174 name: "Targmax"
215 type_attr: "Targmax"
244 name: "Targmax"
286 type_attr: "Targmax"
315 name: "Targmax"
[all …]
|
| D | MaxPoolGradGradWithArgmax.pbtxt | 13 type_attr: "Targmax"
42 name: "Targmax"
81 type_attr: "Targmax"
110 name: "Targmax"
151 type_attr: "Targmax"
180 name: "Targmax"
222 type_attr: "Targmax"
251 name: "Targmax"
293 type_attr: "Targmax"
329 name: "Targmax"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | MaxPoolWithArgmax.pbtxt | 13 type_attr: "Targmax"
28 name: "Targmax"
76 type_attr: "Targmax"
91 name: "Targmax"
143 type_attr: "Targmax"
158 name: "Targmax"
212 type_attr: "Targmax"
227 name: "Targmax"
282 type_attr: "Targmax"
297 name: "Targmax"
[all …]
|
| D | MaxPoolGradWithArgmax.pbtxt | 13 type_attr: "Targmax"
42 name: "Targmax"
77 type_attr: "Targmax"
106 name: "Targmax"
145 type_attr: "Targmax"
174 name: "Targmax"
215 type_attr: "Targmax"
244 name: "Targmax"
286 type_attr: "Targmax"
315 name: "Targmax"
[all …]
|
| D | MaxPoolGradGradWithArgmax.pbtxt | 13 type_attr: "Targmax"
42 name: "Targmax"
81 type_attr: "Targmax"
110 name: "Targmax"
151 type_attr: "Targmax"
180 name: "Targmax"
222 type_attr: "Targmax"
251 name: "Targmax"
293 type_attr: "Targmax"
329 name: "Targmax"
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | maxpooling_op.cc | 62 template <typename Device, typename T, typename Targmax>
74 context, (std::is_same<Targmax, int64_t>::value), in SpatialMaxPoolWithArgMaxHelper()
84 typedef Eigen::Map<Eigen::Matrix<Targmax, Eigen::Dynamic, Eigen::Dynamic>> in SpatialMaxPoolWithArgMaxHelper()
94 output_arg_max->flat<Targmax>().data(), params.depth, in SpatialMaxPoolWithArgMaxHelper()
161 Targmax& out_arg_max_ref = in SpatialMaxPoolWithArgMaxHelper()
945 template <typename Device, typename T, typename Targmax>
948 template <typename T, typename Targmax>
949 struct LaunchMaxPoolingWithArgmax<CPUDevice, T, Targmax> {
954 SpatialMaxPoolWithArgMaxHelper<CPUDevice, T, Targmax>( in launch()
960 template <typename Device, typename T, typename Targmax>
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/ |
| D | unique_output_name.mlir | 6 …ntrol = tf_executor.island wraps "tf.MaxPoolWithArgmax"(%arg0) {T = f32, Targmax = i32, include_ba…
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_ops.py | 5190 Targmax=output_dtype,
5203 Targmax=None, argument
5211 Targmax = deprecated_argument_lookup(
5212 "output_dtype", output_dtype, "Targmax", Targmax)
5213 if Targmax is None:
5214 Targmax = dtypes.int64
5220 Targmax=Targmax,
|
| /external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| D | pooling_ops_test.py | 1011 Targmax=config.Targmax,
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.nn.pbtxt | 289 …argspec: "args=[\'input\', \'ksize\', \'strides\', \'padding\', \'data_format\', \'Targmax\', \'na…
|
| D | tensorflow.raw_ops.pbtxt | 2561 …argspec: "args=[\'input\', \'ksize\', \'strides\', \'padding\', \'Targmax\', \'include_batch_in_in…
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 36989 type_attr: "Targmax"
37018 name: "Targmax"
37057 type_attr: "Targmax"
37086 name: "Targmax"
37127 type_attr: "Targmax"
37156 name: "Targmax"
37198 type_attr: "Targmax"
37227 name: "Targmax"
37269 type_attr: "Targmax"
37305 name: "Targmax"
[all …]
|
| D | ops_history.v0.pbtxt | 17166 type_attr: "Targmax"
17195 name: "Targmax"
17217 type_attr: "Targmax"
17246 name: "Targmax"
17281 type_attr: "Targmax"
17296 name: "Targmax"
17331 type_attr: "Targmax"
17346 name: "Targmax"
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 27205 type_attr: "Targmax"
27241 name: "Targmax"
27356 type_attr: "Targmax"
27392 name: "Targmax"
27499 type_attr: "Targmax"
27514 name: "Targmax"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 2561 …argspec: "args=[\'input\', \'ksize\', \'strides\', \'padding\', \'Targmax\', \'include_batch_in_in…
|