| /external/tensorflow/tensorflow/core/kernels/ |
| D | maxpooling_op_gpu.cu.cc | 73 const bool include_batch_in_index) { in MaxPoolForwardNCHW() argument
93 maxidx = include_batch_in_index ? idx + offset : idx; in MaxPoolForwardNCHW()
150 const bool include_batch_in_index) { in MaxPoolForwardNHWC() argument
171 maxidx = include_batch_in_index ? idx + offset : idx; in MaxPoolForwardNHWC()
208 const bool include_batch_in_index) { in MaxPoolBackward() argument
211 include_batch_in_index ? 0 : (index / top_offset) * bottom_offset; in MaxPoolBackward()
338 const bool include_batch_in_index) { in MaxPoolGradBackward() argument
341 include_batch_in_index ? 0 : (index / bottom_offset) * top_offset; in MaxPoolGradBackward()
380 const bool include_batch_in_index) { in operator ()() argument
391 pad_t, pad_l, top_data, mask, include_batch_in_index)); in operator ()()
[all …]
|
| D | maxpooling_op_gpu.h | 43 const bool include_batch_in_index);
60 const Eigen::GpuDevice& d, const bool include_batch_in_index);
68 const Eigen::GpuDevice& d, const bool include_batch_in_index);
|
| D | maxpooling_op.cc | 65 const PoolParameters& params, const bool include_batch_in_index) { in SpatialMaxPoolWithArgMaxHelper() argument
68 context, include_batch_in_index, in SpatialMaxPoolWithArgMaxHelper()
110 include_batch_in_index](int64 start, int64 limit) { in SpatialMaxPoolWithArgMaxHelper()
164 if (include_batch_in_index) { in SpatialMaxPoolWithArgMaxHelper()
917 bool propagate_nans, bool include_batch_in_index) { in launch()
921 params, include_batch_in_index); in launch()
992 Tensor* grad_out, const bool include_batch_in_index) { in launch()
996 auto shard = [&grad_in, &argmax, &grad_out, include_batch_in_index]( in launch()
1018 if (!include_batch_in_index) { in launch()
1377 bool propagate_nans, bool include_batch_in_index) { in launch()
[all …]
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_MaxPoolWithArgmax.pbtxt | 41 name: "include_batch_in_index"
50 `(y * width + x) * channels + c` if `include_batch_in_index` is False;
51 `((b * height + y) * width + x) * channels + c` if `include_batch_in_index` is True.
|
| D | api_def_MaxPoolGradGradWithArgmax.pbtxt | 48 name: "include_batch_in_index"
|
| D | api_def_MaxPoolGradWithArgmax.pbtxt | 49 name: "include_batch_in_index"
|
| /external/tensorflow/tensorflow/lite/kernels/perception/ |
| D | max_pool_with_argmax.cc | 109 bool include_batch_in_index; member
119 op_data->include_batch_in_index = m[kIncludeBatchStr].AsBool(); in Init()
173 context, !op_data->include_batch_in_index, in Prepare()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | pooling_ops_test.py | 1026 include_batch_in_index=config.include_batch_in_index)
1063 include_batch_in_index=config.include_batch_in_index)
1103 include_batch_in_index=config.include_batch_in_index)
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | MaxPoolGradGradWithArgmax.pbtxt | 322 name: "include_batch_in_index"
|
| D | MaxPoolWithArgmax.pbtxt | 390 name: "include_batch_in_index"
|
| D | MaxPoolGradWithArgmax.pbtxt | 386 name: "include_batch_in_index"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | MaxPoolGradGradWithArgmax.pbtxt | 322 name: "include_batch_in_index"
|
| D | MaxPoolWithArgmax.pbtxt | 390 name: "include_batch_in_index"
|
| D | MaxPoolGradWithArgmax.pbtxt | 386 name: "include_batch_in_index"
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_ops.py | 4777 include_batch_in_index=False, argument
4834 include_batch_in_index=include_batch_in_index,
4849 include_batch_in_index=False): argument
4863 include_batch_in_index=include_batch_in_index,
|
| D | nn_grad.py | 719 include_batch_in_index=op.get_attr("include_batch_in_index"))
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.nn.pbtxt | 233 …, \'strides\', \'padding\', \'data_format\', \'output_dtype\', \'include_batch_in_index\', \'name\…
|
| D | tensorflow.raw_ops.pbtxt | 2481 …\', \'grad\', \'argmax\', \'ksize\', \'strides\', \'padding\', \'include_batch_in_index\', \'name\…
2489 …\', \'grad\', \'argmax\', \'ksize\', \'strides\', \'padding\', \'include_batch_in_index\', \'name\…
2497 …=[\'input\', \'ksize\', \'strides\', \'padding\', \'Targmax\', \'include_batch_in_index\', \'name\…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.nn.pbtxt | 277 …g\', \'data_format\', \'Targmax\', \'name\', \'output_dtype\', \'include_batch_in_index\'], vararg…
|
| D | tensorflow.raw_ops.pbtxt | 2481 …\', \'grad\', \'argmax\', \'ksize\', \'strides\', \'padding\', \'include_batch_in_index\', \'name\…
2489 …\', \'grad\', \'argmax\', \'ksize\', \'strides\', \'padding\', \'include_batch_in_index\', \'name\…
2497 …=[\'input\', \'ksize\', \'strides\', \'padding\', \'Targmax\', \'include_batch_in_index\', \'name\…
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 24685 name: "include_batch_in_index"
24836 name: "include_batch_in_index"
24988 name: "include_batch_in_index"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 37298 name: "include_batch_in_index"
38008 name: "include_batch_in_index"
38625 name: "include_batch_in_index"
|