Home
last modified time | relevance | path

Searched refs:include_batch_in_index (Results 1 – 25 of 25) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dmaxpooling_op_gpu.cu.cc73 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 …]
H A Dmaxpooling_op_gpu.h43 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);
H A Dmaxpooling_op.cc66 const PoolParameters& params, const bool include_batch_in_index) { in SpatialMaxPoolWithArgMaxHelper() argument
69 context, include_batch_in_index, in SpatialMaxPoolWithArgMaxHelper()
112 include_batch_in_index](int64_t start, int64_t limit) { in SpatialMaxPoolWithArgMaxHelper()
166 if (include_batch_in_index) { in SpatialMaxPoolWithArgMaxHelper()
952 bool propagate_nans, bool include_batch_in_index) { in launch()
956 params, include_batch_in_index); in launch()
1031 Tensor* grad_out, const bool include_batch_in_index) { in launch()
1035 auto shard = [&grad_in, &argmax, &grad_out, include_batch_in_index]( in launch()
1061 if (!include_batch_in_index) { in launch()
1451 bool propagate_nans, bool include_batch_in_index) { in launch()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/core/api_def/base_api/
H A Dapi_def_MaxPoolWithArgmax.pbtxt41 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.
H A Dapi_def_MaxPoolGradGradWithArgmax.pbtxt48 name: "include_batch_in_index"
H A Dapi_def_MaxPoolGradWithArgmax.pbtxt49 name: "include_batch_in_index"
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/perception/
H A Dmax_pool_with_argmax.cc112 bool include_batch_in_index; member
122 op_data->include_batch_in_index = m[kIncludeBatchStr].AsBool(); in Init()
176 context, !op_data->include_batch_in_index, in Prepare()
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/
H A Dpooling_ops_test.py1013 include_batch_in_index=config.include_batch_in_index)
1050 include_batch_in_index=config.include_batch_in_index)
1082 include_batch_in_index=False)
1107 include_batch_in_index=False)
1145 include_batch_in_index=config.include_batch_in_index)
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/
H A Dunique_output_name.mlir6 …and wraps "tf.MaxPoolWithArgmax"(%arg0) {T = f32, Targmax = i32, include_batch_in_index = false, k…
/aosp_15_r20/external/tensorflow/tensorflow/security/advisory/
H A Dtfsa-2021-062.md21 padding='SAME', include_batch_in_index=False)
H A Dtfsa-2021-057.md21 padding='SAME', include_batch_in_index=False)
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dnn_ops.py5131 include_batch_in_index=False, argument
5191 include_batch_in_index=include_batch_in_index,
5206 include_batch_in_index=False): argument
5221 include_batch_in_index=include_batch_in_index,
H A Dnn_grad.py713 include_batch_in_index=op.get_attr("include_batch_in_index"))
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
H A DMaxPoolGradGradWithArgmax.pbtxt322 name: "include_batch_in_index"
H A DMaxPoolWithArgmax.pbtxt390 name: "include_batch_in_index"
H A DMaxPoolGradWithArgmax.pbtxt386 name: "include_batch_in_index"
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
H A DMaxPoolGradGradWithArgmax.pbtxt322 name: "include_batch_in_index"
H A DMaxPoolWithArgmax.pbtxt390 name: "include_batch_in_index"
H A DMaxPoolGradWithArgmax.pbtxt386 name: "include_batch_in_index"
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/
H A Dtensorflow.nn.pbtxt245 …, \'strides\', \'padding\', \'data_format\', \'output_dtype\', \'include_batch_in_index\', \'name\…
H A Dtensorflow.raw_ops.pbtxt2545 …\', \'grad\', \'argmax\', \'ksize\', \'strides\', \'padding\', \'include_batch_in_index\', \'name\…
2553 …\', \'grad\', \'argmax\', \'ksize\', \'strides\', \'padding\', \'include_batch_in_index\', \'name\…
2561 …=[\'input\', \'ksize\', \'strides\', \'padding\', \'Targmax\', \'include_batch_in_index\', \'name\…
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/
H A Dtensorflow.nn.pbtxt289 …g\', \'data_format\', \'Targmax\', \'name\', \'output_dtype\', \'include_batch_in_index\'], vararg…
H A Dtensorflow.raw_ops.pbtxt2545 …\', \'grad\', \'argmax\', \'ksize\', \'strides\', \'padding\', \'include_batch_in_index\', \'name\…
2553 …\', \'grad\', \'argmax\', \'ksize\', \'strides\', \'padding\', \'include_batch_in_index\', \'name\…
2561 …=[\'input\', \'ksize\', \'strides\', \'padding\', \'Targmax\', \'include_batch_in_index\', \'name\…
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Dops.pbtxt27234 name: "include_batch_in_index"
27385 name: "include_batch_in_index"
27537 name: "include_batch_in_index"
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/compat/
H A Dops_history.v2.pbtxt37298 name: "include_batch_in_index"
38008 name: "include_batch_in_index"
38625 name: "include_batch_in_index"