| /external/tensorflow/tensorflow/core/kernels/ |
| D | maxpooling_op_gpu.cu.cc | 71 dtype* top_data, int64* mask, const bool include_batch_in_index) { in MaxPoolForwardNCHW() argument
91 maxidx = include_batch_in_index ? idx + offset : idx; in MaxPoolForwardNCHW()
145 dtype* top_data, int64* mask, const bool include_batch_in_index) { in MaxPoolForwardNHWC() argument
166 maxidx = include_batch_in_index ? idx + offset : idx; in MaxPoolForwardNHWC()
243 const bool include_batch_in_index) { in MaxPoolBackward() argument
246 include_batch_in_index ? 0 : (index / top_offset) * bottom_offset; in MaxPoolBackward()
370 const bool include_batch_in_index) { in MaxPoolGradBackward() argument
373 include_batch_in_index ? 0 : (index / bottom_offset) * top_offset; in MaxPoolGradBackward()
409 const bool include_batch_in_index) { in operator ()() argument
419 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);
78 const Eigen::GpuDevice& d, const bool include_batch_in_index);
|
| D | maxpooling_op.cc | 61 const PoolParameters& params, const bool include_batch_in_index) { in SpatialMaxPoolWithArgMaxHelper() argument
64 context, include_batch_in_index, in SpatialMaxPoolWithArgMaxHelper()
102 include_batch_in_index](int64 start, int64 limit) { in SpatialMaxPoolWithArgMaxHelper()
155 if (include_batch_in_index) { in SpatialMaxPoolWithArgMaxHelper()
890 bool propagate_nans, bool include_batch_in_index) { in launch()
894 include_batch_in_index); in launch()
960 Tensor* grad_out, const bool include_batch_in_index) { in launch()
964 auto shard = [&grad_in, &argmax, &grad_out, include_batch_in_index]( in launch()
986 if (!include_batch_in_index) { in launch()
1321 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/python/kernel_tests/ |
| D | pooling_ops_test.py | 862 include_batch_in_index=config.include_batch_in_index)
899 include_batch_in_index=config.include_batch_in_index)
939 include_batch_in_index=config.include_batch_in_index)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_ops.py | 3834 include_batch_in_index=False, argument
3891 include_batch_in_index=include_batch_in_index,
3905 include_batch_in_index=False): argument
3919 include_batch_in_index=include_batch_in_index,
|
| D | nn_grad.py | 709 include_batch_in_index=op.get_attr("include_batch_in_index"))
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.nn.pbtxt | 225 …, \'strides\', \'padding\', \'data_format\', \'output_dtype\', \'include_batch_in_index\', \'name\…
|
| D | tensorflow.raw_ops.pbtxt | 1981 …\', \'grad\', \'argmax\', \'ksize\', \'strides\', \'padding\', \'include_batch_in_index\', \'name\…
1989 …\', \'grad\', \'argmax\', \'ksize\', \'strides\', \'padding\', \'include_batch_in_index\', \'name\…
1997 …=[\'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 | 1981 …\', \'grad\', \'argmax\', \'ksize\', \'strides\', \'padding\', \'include_batch_in_index\', \'name\…
1989 …\', \'grad\', \'argmax\', \'ksize\', \'strides\', \'padding\', \'include_batch_in_index\', \'name\…
1997 …=[\'input\', \'ksize\', \'strides\', \'padding\', \'Targmax\', \'include_batch_in_index\', \'name\…
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 19208 name: "include_batch_in_index"
19359 name: "include_batch_in_index"
19511 name: "include_batch_in_index"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 37961 name: "include_batch_in_index"
38671 name: "include_batch_in_index"
39288 name: "include_batch_in_index"
|
| D | ops_history.v1.pbtxt | 38399 name: "include_batch_in_index"
39109 name: "include_batch_in_index"
39726 name: "include_batch_in_index"
|