Home
last modified time | relevance | path

Searched refs:tensor_in_shape (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dpooling_ops_common.cc50 const TensorShape& tensor_in_shape) { in PoolParameters() argument
55 GetTensorSpatialDims(tensor_in_shape.dims(), data_format) == 2, in PoolParameters()
58 tensor_in_shape.dims(), " ", data_format)); in PoolParameters()
61 depth = GetTensorDim(tensor_in_shape, data_format, 'C') * in PoolParameters()
63 tensor_in_cols = GetTensorDim(tensor_in_shape, data_format, 'W'); in PoolParameters()
64 tensor_in_rows = GetTensorDim(tensor_in_shape, data_format, 'H'); in PoolParameters()
65 tensor_in_batch = GetTensorDim(tensor_in_shape, data_format, 'N'); in PoolParameters()
290 const TensorShape& tensor_in_shape, bool propagate_nans) { in Compute() argument
298 context->allocate_output(0, tensor_in_shape, &input_backprop)); in Compute()
299 if (tensor_in_shape.num_elements() == 0) { in Compute()
[all …]
Davgpooling_op.cc243 const Tensor& tensor_in_shape = context->input(0); in Compute() local
248 tensor_in_shape.dims() == 1 && tensor_in_shape.NumElements() == 4, in Compute()
260 auto shape_vec = tensor_in_shape.vec<int32>(); in Compute()
261 for (int64 i = 0; i < tensor_in_shape.NumElements(); ++i) { in Compute()
402 const Tensor& tensor_in_shape = context->input(0); in Compute() local
407 tensor_in_shape.dims() == 1 && tensor_in_shape.NumElements() == 4, in Compute()
415 auto shape_vec = tensor_in_shape.vec<int32>(); in Compute()
416 for (int64 i = 0; i < tensor_in_shape.NumElements(); ++i) { in Compute()
482 const Tensor& tensor_in_shape = context->input(0); in Compute() local
487 tensor_in_shape.dims() == 1 && tensor_in_shape.NumElements() == 4, in Compute()
[all …]
Dmkl_pooling_ops_common.cc228 const TensorShape& tensor_in_shape) { in Init() argument
231 tensor_in_shape.dims() == 4 || tensor_in_shape.dims() == 5, in Init()
234 depth = GetTensorDim(tensor_in_shape, data_format, 'C'); in Init()
235 if (tensor_in_shape.dims() == 4) { in Init()
237 tensor_in_cols = GetTensorDim(tensor_in_shape, data_format, 'W'); in Init()
238 tensor_in_rows = GetTensorDim(tensor_in_shape, data_format, 'H'); in Init()
241 tensor_in_planes = GetTensorDim(tensor_in_shape, data_format, '0'); in Init()
242 tensor_in_rows = GetTensorDim(tensor_in_shape, data_format, '1'); in Init()
243 tensor_in_cols = GetTensorDim(tensor_in_shape, data_format, '2'); in Init()
245 tensor_in_batch = GetTensorDim(tensor_in_shape, data_format, 'N'); in Init()
Dpooling_ops_3d.cc57 const TensorShape& tensor_in_shape) { in Pool3dParameters() argument
59 OP_REQUIRES(context, tensor_in_shape.dims() == 5, in Pool3dParameters()
63 depth = GetTensorDim(tensor_in_shape, data_format, 'C'); in Pool3dParameters()
64 tensor_in_planes = GetTensorDim(tensor_in_shape, data_format, '0'); in Pool3dParameters()
65 tensor_in_rows = GetTensorDim(tensor_in_shape, data_format, '1'); in Pool3dParameters()
66 tensor_in_cols = GetTensorDim(tensor_in_shape, data_format, '2'); in Pool3dParameters()
67 tensor_in_batch = GetTensorDim(tensor_in_shape, data_format, 'N'); in Pool3dParameters()
383 const TensorShape& tensor_in_shape, in launch()
391 std::array<int64, 3> input_size = {{tensor_in_shape.dim_size(3), in launch()
392 tensor_in_shape.dim_size(2), in launch()
[all …]
Dcudnn_pooling_gpu.cc123 const Tensor& out_backprop, const TensorShape& tensor_in_shape, in Compute() argument
130 const int64 in_batch = GetTensorDim(tensor_in_shape, data_format, 'N'); in Compute()
131 const int64 in_features = GetTensorDim(tensor_in_shape, data_format, 'C'); in Compute()
137 ShapeFromFormat(FORMAT_NCHW, tensor_in_shape, data_format); in Compute()
208 dim_i, GetTensorDim(tensor_in_shape, data_format, '2' - i)); in Compute()
Dmkl_avgpooling_op.cc245 const Tensor& tensor_in_shape = MklGetInput(context, 0); in Compute() local
252 auto shape_vec = tensor_in_shape.vec<int32>(); in Compute()
253 for (int64 i = 0; i < tensor_in_shape.NumElements(); ++i) { in Compute()
353 const Tensor& tensor_in_shape = MklGetInput(context, 0); in MklCreateLayoutsAndPrimitives() local
362 tensor_in_shape.dims() == 1 && tensor_in_shape.NumElements() == 4, in MklCreateLayoutsAndPrimitives()
673 const Tensor& tensor_in_shape, in SanityCheckInputs() argument
680 tensor_in_shape.dims() == 1 && tensor_in_shape.NumElements() == 4, in SanityCheckInputs()
Dcudnn_pooling_gpu.h61 const TensorShape& tensor_in_shape,
Dpooling_ops_3d.h45 const TensorShape& tensor_in_shape);
Dpooling_ops_common_gpu.h63 const TensorShape& tensor_in_shape, bool propagate_nans);
Dpooling_ops_3d_sycl.h719 const TensorShape& tensor_in_shape,
727 const int batch = GetTensorDim(tensor_in_shape, data_format, 'N');
728 const int in_planes = GetTensorDim(tensor_in_shape, data_format, '0');
729 const int in_rows = GetTensorDim(tensor_in_shape, data_format, '1');
730 const int in_cols = GetTensorDim(tensor_in_shape, data_format, '2');
731 const int depth = GetTensorDim(tensor_in_shape, data_format, 'C');
Dmaxpooling_op.cc326 const Tensor& out_backprop, const TensorShape& tensor_in_shape) { in MaxPoolingBackwardCustomKernel() argument
329 {0}, 0, tensor_in_shape, &output)); in MaxPoolingBackwardCustomKernel()
332 padding, FORMAT_NHWC, tensor_in_shape}; in MaxPoolingBackwardCustomKernel()
Dpooling_ops_common.h45 TensorFormat data_format, const TensorShape& tensor_in_shape);
Dmkl_pooling_ops_common.h426 TensorFormat data_format, const TensorShape& tensor_in_shape);
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dpooling_ops.cc308 const TensorShape tensor_in_shape = ctx->InputShape(0); in Compile() local
313 OP_REQUIRES(ctx, tensor_in_shape.dims() == num_dims(), in Compile()
515 const TensorShape tensor_in_shape = ctx->InputShape(0); in Compile() local
520 OP_REQUIRES(ctx, tensor_in_shape.dims() == num_dims(), in Compile()