Searched refs:Targmax (Results 1 – 15 of 15) sorted by relevance
| /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/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/kernels/ |
| D | maxpooling_op.cc | 61 template <typename Device, typename T, typename Targmax>
73 context, (std::is_same<Targmax, int64>::value), in SpatialMaxPoolWithArgMaxHelper()
82 typedef Eigen::Map<Eigen::Matrix<Targmax, Eigen::Dynamic, Eigen::Dynamic>> in SpatialMaxPoolWithArgMaxHelper()
92 output_arg_max->flat<Targmax>().data(), params.depth, in SpatialMaxPoolWithArgMaxHelper()
159 Targmax& out_arg_max_ref = in SpatialMaxPoolWithArgMaxHelper()
910 template <typename Device, typename T, typename Targmax>
913 template <typename T, typename Targmax>
914 struct LaunchMaxPoolingWithArgmax<CPUDevice, T, Targmax> {
919 SpatialMaxPoolWithArgMaxHelper<CPUDevice, T, Targmax>( in launch()
925 template <typename Device, typename T, typename Targmax>
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_ops.py | 4833 Targmax=output_dtype,
4846 Targmax=None, argument
4853 Targmax = deprecated_argument_lookup(
4854 "output_dtype", output_dtype, "Targmax", Targmax)
4855 if Targmax is None:
4856 Targmax = dtypes.int64
4862 Targmax=Targmax,
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | pooling_ops_test.py | 1024 Targmax=config.Targmax,
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.nn.pbtxt | 277 …argspec: "args=[\'input\', \'ksize\', \'strides\', \'padding\', \'data_format\', \'Targmax\', \'na…
|
| D | tensorflow.raw_ops.pbtxt | 2497 …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 | 24656 type_attr: "Targmax"
24692 name: "Targmax"
24807 type_attr: "Targmax"
24843 name: "Targmax"
24950 type_attr: "Targmax"
24965 name: "Targmax"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 2497 …argspec: "args=[\'input\', \'ksize\', \'strides\', \'padding\', \'Targmax\', \'include_batch_in_in…
|