Home
last modified time | relevance | path

Searched refs:tensor_in_batch (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmaxpooling_op.cc79 params.tensor_in_cols * params.tensor_in_rows * params.tensor_in_batch); in SpatialMaxPoolWithArgMaxHelper()
82 params.out_width * params.out_height * params.tensor_in_batch); in SpatialMaxPoolWithArgMaxHelper()
85 params.out_width * params.out_height * params.tensor_in_batch); in SpatialMaxPoolWithArgMaxHelper()
202 params.tensor_in_batch, shard_cost, shard); in SpatialMaxPoolWithArgMaxHelper()
338 tensor_in->flat<T>().data(), params.tensor_in_batch, in MaxPoolingBackwardCustomKernel()
550 params.tensor_in_cols * params.tensor_in_rows * params.tensor_in_batch); in SpatialMaxPoolGradGrad()
553 params.out_width * params.out_height * params.tensor_in_batch); in SpatialMaxPoolGradGrad()
556 params.tensor_in_cols * params.tensor_in_rows * params.tensor_in_batch); in SpatialMaxPoolGradGrad()
559 params.out_width * params.out_height * params.tensor_in_batch); in SpatialMaxPoolGradGrad()
641 params.tensor_in_batch, shard_cost, shard); in SpatialMaxPoolGradGrad()
[all …]
Dpooling_ops_common.h54 int tensor_in_batch; member
178 params.tensor_in_batch); in SpatialMaxPool()
181 params.out_width * params.out_height * params.tensor_in_batch); in SpatialMaxPool()
254 params.tensor_in_batch, shard_cost, shard); in SpatialMaxPool()
274 params.tensor_in_batch, params.tensor_in_rows, params.tensor_in_cols,
425 params.tensor_in_batch);
428 params.out_width * params.out_height * params.tensor_in_batch);
501 params.tensor_in_batch, shard_cost, shard);
603 params.tensor_in_batch, work_unit_cost, shard);
Dpooling_ops_common.cc65 tensor_in_batch = GetTensorDim(tensor_in_shape, data_format, 'N'); in PoolParameters()
119 return ShapeFromFormat(data_format, tensor_in_batch, out_height, out_width, in forward_output_shape()
124 {tensor_in_batch, tensor_in_rows, tensor_in_cols, out_depth}); in forward_output_shape()
174 int batch_size = params.tensor_in_batch; in Compute()
387 orig_output_desc.set_count(params.tensor_in_batch) in Compute()
394 orig_input_desc.set_count(params.tensor_in_batch) in Compute()
Dmkl_pooling_ops_common.cc245 tensor_in_batch = GetTensorDim(tensor_in_shape, data_format, 'N'); in Init()
261 tensor_in_batch = mklInputShape->GetSizes()[3]; in Init()
278 tensor_in_batch = mklInputShape->GetDimension('N'); in Init()
285 tensor_in_batch = mklInputShape->GetDimension3D('N'); in Init()
411 mkl_params->in_sizes[3] = params.tensor_in_batch; in ExtractMklOpParams()
419 mkl_params->out_sizes[3] = params.tensor_in_batch; in ExtractMklOpParams()
Dpooling_ops_3d.cc67 tensor_in_batch = GetTensorDim(tensor_in_shape, data_format, 'N'); in Pool3dParameters()
96 return ShapeFromFormat(data_format, tensor_in_batch, in forward_output_shape()
560 params.tensor_in_batch); in launch()
563 params.out_height * params.tensor_in_batch); in launch()
567 params.tensor_in_rows * params.tensor_in_batch); in launch()
571 params.tensor_in_batch); in launch()
651 params.tensor_in_batch, shard_cost, shard); in launch()
810 tensor_out.flat<T>().data(), params.tensor_in_batch, params.out_plane, in launch()
Dpooling_ops_3d.h55 int tensor_in_batch; member
Dmkl_pooling_ops_common.h370 int tensor_in_batch; member
401 tensor_in_batch(0), in MklPoolParameters()
494 *output_dims_mkl_order = {mkl_pool_params.tensor_in_batch, in GetOutputDims()
500 *output_dims_mkl_order = {mkl_pool_params.tensor_in_batch, in GetOutputDims()
565 output_dims_NHWC_order = {pool_params->tensor_in_batch, in AllocateEmptyOutputTensor()
Dpooling_ops_3d_sycl.h69 batch_(params.tensor_in_batch), in SYCL3DPoolParams()