Home
last modified time | relevance | path

Searched defs:in_rows (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dpad_test.cc341 constexpr int64_t in_rows = 35; in XLA_TEST_P() local
363 constexpr int64_t in_rows = 129; in XLA_TEST_P() local
390 constexpr int64_t in_rows = 129; in XLA_TEST_P() local
418 constexpr int64_t in_rows = 8; in XLA_TEST_P() local
/external/ComputeLibrary/tests/validation/reference/
DGEMMInterleave4x4.h41 const int32_t in_rows = in.shape().y(); in gemm_interleave_4x4() local
/external/tensorflow/tensorflow/core/kernels/
Ddepthwise_conv_op.h28 int in_rows; member
Ddeep_conv2d.h70 int in_rows; member
Dfused_batch_norm_op.cc132 const int64_t in_rows = GetTensorDim(x_input, tensor_format, 'H'); in operator ()() local
273 const int64_t in_rows = GetTensorDim(x_input, tensor_format, 'H'); in operator ()() local
374 const int64_t in_rows = GetTensorDim(x_input, tensor_format, 'H'); in operator ()() local
1299 int64_t in_rows = GetTensorDim(x, tensor_format_, '1'); in ComputeWithReservedSpace() local
1544 int64_t in_rows = GetTensorDim(x, tensor_format_, '1'); in ComputeWithReservedSpace() local
Deigen_spatial_convolutions_test.cc670 const int in_rows = 8; in TEST() local
731 const int in_rows = 8; in TEST() local
793 const int in_rows = 5; in TEST() local
845 const int in_rows = 5; in TEST() local
899 const int in_rows = 8; in TEST() local
965 const int in_rows = 8; in TEST() local
1031 const int in_rows = 7; in TEST() local
1086 const int in_rows = 7; in TEST() local
1142 const int in_rows = 7; in TEST() local
1212 const int in_rows = 7; in TEST() local
Davgpooling_op.cc309 const int64_t in_rows = output_shape.dim_size(1); in Compute() local
574 const int64 in_rows = output_shape.dim_size(1); in Compute() local
Dpooling_ops_common.h234 const int32_t in_rows = params.tensor_in_rows; in SpatialMaxPool() local
Dfractional_avg_pool_op.cc297 const int64_t in_rows = orig_input_tensor_shape_flat(1); in Compute() local
Dconv_ops_3d.cc242 int64_t in_rows = GetTensorDim(input, data_format, '1'); in launch() local
Dmaxpooling_op.cc114 const int32_t in_rows = params.tensor_in_rows; in SpatialMaxPoolWithArgMaxHelper() local
616 const int32_t in_rows = params.tensor_in_rows; in SpatialMaxPoolGradGrad() local
Dpooling_ops_3d.cc599 const int32_t in_rows = params.tensor_in_rows; in launch() local
Dconv_ops.cc822 int64_t in_rows = GetTensorDim(input, data_format, 'H'); in operator ()() local
/external/tensorflow/tensorflow/core/kernels/image/
Dextract_volume_patches_op.cc80 const int in_rows = input.dim_size(2); in Compute() local
Dextract_image_patches_op.cc73 const int in_rows = input.dim_size(1); in Compute() local
/external/pytorch/aten/src/ATen/native/cpu/
DDepthwiseConvKernel.cpp26 int64_t in_rows; member
/external/tensorflow/tensorflow/core/ops/
Darray_ops.cc2674 auto in_rows = c->Value(in_rows_dim); in __anon847f0b683902() local
2780 auto in_rows = c->Value(in_rows_dim); in __anon847f0b683a02() local
Dnn_ops.cc1064 auto in_rows = c->Value(in_rows_dim); in __anoncb4947e51702() local