/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | pad_test.cc | 341 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/ |
D | GEMMInterleave4x4.h | 41 const int32_t in_rows = in.shape().y(); in gemm_interleave_4x4() local
|
/external/tensorflow/tensorflow/core/kernels/ |
D | depthwise_conv_op.h | 28 int in_rows; member
|
D | deep_conv2d.h | 70 int in_rows; member
|
D | fused_batch_norm_op.cc | 132 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
|
D | eigen_spatial_convolutions_test.cc | 670 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
|
D | avgpooling_op.cc | 309 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
|
D | pooling_ops_common.h | 234 const int32_t in_rows = params.tensor_in_rows; in SpatialMaxPool() local
|
D | fractional_avg_pool_op.cc | 297 const int64_t in_rows = orig_input_tensor_shape_flat(1); in Compute() local
|
D | conv_ops_3d.cc | 242 int64_t in_rows = GetTensorDim(input, data_format, '1'); in launch() local
|
D | maxpooling_op.cc | 114 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
|
D | pooling_ops_3d.cc | 599 const int32_t in_rows = params.tensor_in_rows; in launch() local
|
D | conv_ops.cc | 822 int64_t in_rows = GetTensorDim(input, data_format, 'H'); in operator ()() local
|
/external/tensorflow/tensorflow/core/kernels/image/ |
D | extract_volume_patches_op.cc | 80 const int in_rows = input.dim_size(2); in Compute() local
|
D | extract_image_patches_op.cc | 73 const int in_rows = input.dim_size(1); in Compute() local
|
/external/pytorch/aten/src/ATen/native/cpu/ |
D | DepthwiseConvKernel.cpp | 26 int64_t in_rows; member
|
/external/tensorflow/tensorflow/core/ops/ |
D | array_ops.cc | 2674 auto in_rows = c->Value(in_rows_dim); in __anon847f0b683902() local 2780 auto in_rows = c->Value(in_rows_dim); in __anon847f0b683a02() local
|
D | nn_ops.cc | 1064 auto in_rows = c->Value(in_rows_dim); in __anoncb4947e51702() local
|