Searched refs:tensor_in_batch (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | maxpooling_op.cc | 89 params.tensor_in_cols * params.tensor_in_rows * params.tensor_in_batch); in SpatialMaxPoolWithArgMaxHelper()
92 params.out_width * params.out_height * params.tensor_in_batch); in SpatialMaxPoolWithArgMaxHelper()
95 params.out_width * params.out_height * params.tensor_in_batch); in SpatialMaxPoolWithArgMaxHelper()
215 params.tensor_in_batch, shard_cost, shard); in SpatialMaxPoolWithArgMaxHelper()
582 params.tensor_in_cols * params.tensor_in_rows * params.tensor_in_batch); in SpatialMaxPoolGradGrad()
585 params.out_width * params.out_height * params.tensor_in_batch); in SpatialMaxPoolGradGrad()
588 params.tensor_in_cols * params.tensor_in_rows * params.tensor_in_batch); in SpatialMaxPoolGradGrad()
591 params.out_width * params.out_height * params.tensor_in_batch); in SpatialMaxPoolGradGrad()
673 params.tensor_in_batch, shard_cost, shard); in SpatialMaxPoolGradGrad()
779 params.tensor_in_batch, params.out_height, params.out_width, in Compute()
[all …]
|
| H A D | pooling_ops_common.h | 62 int tensor_in_batch; member
214 params.tensor_in_batch); in SpatialMaxPool()
217 params.out_width * params.out_height * params.tensor_in_batch); in SpatialMaxPool()
291 params.tensor_in_batch, shard_cost, shard); in SpatialMaxPool()
313 params.tensor_in_batch, params.tensor_in_rows, params.tensor_in_cols,
479 params.tensor_in_batch);
482 params.out_width * params.out_height * params.tensor_in_batch);
556 params.tensor_in_batch, shard_cost, shard);
661 params.tensor_in_batch, work_unit_cost, shard);
|
| H A D | pooling_ops_common.cc | 138 tensor_in_batch = GetTensorDim(tensor_in_shape, data_format, 'N'); in PoolParameters()
204 return ShapeFromFormat(data_format, tensor_in_batch, out_height, out_width, in forward_output_shape()
209 {tensor_in_batch, tensor_in_rows, tensor_in_cols, out_depth}); in forward_output_shape()
238 int batch_size = params.tensor_in_batch; in Compute()
582 int batch_size = params.tensor_in_batch; in Compute()
687 orig_output_desc.set_count(params.tensor_in_batch) in Compute()
694 orig_input_desc.set_count(params.tensor_in_batch) in Compute()
|
| H A D | pooling_ops_3d.cc | 62 tensor_in_batch = GetTensorDim(tensor_in_shape, data_format, 'N'); in Pool3dParameters()
91 return ShapeFromFormat(data_format, tensor_in_batch, in forward_output_shape()
579 params.tensor_in_batch); in launch()
582 params.out_height * params.tensor_in_batch); in launch()
586 params.tensor_in_rows * params.tensor_in_batch); in launch()
590 params.tensor_in_batch); in launch()
670 params.tensor_in_batch, shard_cost, shard); in launch()
862 tensor_out.flat<T>().data(), params.tensor_in_batch, params.out_plane, in launch()
|
| H A D | pooling_ops_3d.h | 55 int tensor_in_batch; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
| H A D | mkl_pooling_ops_common.h | 372 int tensor_in_batch; member
403 tensor_in_batch(0), in MklPoolParameters()
499 *output_dims_mkl_order = {mkl_pool_params.tensor_in_batch, in GetOutputDims()
505 *output_dims_mkl_order = {mkl_pool_params.tensor_in_batch, in GetOutputDims()
572 output_dims_order = {pool_params->tensor_in_batch, in AllocateEmptyOutputTensor()
577 output_dims_order = {pool_params->tensor_in_batch, in AllocateEmptyOutputTensor()
|
| H A D | mkl_pooling_ops_common.cc | 250 tensor_in_batch = GetTensorDim(tensor_in_shape, data_format, 'N'); in Init()
267 tensor_in_batch = mklInputShape->GetDimension('N'); in Init()
274 tensor_in_batch = mklInputShape->GetDimension3D('N'); in Init()
|