| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | pad_test.cc | 342 constexpr int64_t in_cols = 35; in XLA_TEST_P() local
364 constexpr int64_t in_cols = 129; in XLA_TEST_P() local
391 constexpr int64_t in_cols = 129; in XLA_TEST_P() local
419 constexpr int64_t in_cols = 11; in XLA_TEST_P() local
|
| /external/ComputeLibrary/tests/validation/reference/ |
| D | GEMMInterleave4x4.h | 42 const int32_t in_cols = in.shape().x(); in gemm_interleave_4x4() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | depthwise_conv_op.h | 29 int in_cols; member
|
| D | deep_conv2d.h | 71 int in_cols; member
|
| D | fused_batch_norm_op.cc | 133 const int64_t in_cols = GetTensorDim(x_input, tensor_format, 'W'); in operator ()() local
274 const int64_t in_cols = GetTensorDim(x_input, tensor_format, 'W'); in operator ()() local
375 const int64_t in_cols = GetTensorDim(x_input, tensor_format, 'W'); in operator ()() local
1300 int64_t in_cols = GetTensorDim(x, tensor_format_, '2'); in ComputeWithReservedSpace() local
1545 int64_t in_cols = GetTensorDim(x, tensor_format_, '2'); in ComputeWithReservedSpace() local
|
| D | eigen_spatial_convolutions_test.cc | 671 const int in_cols = 7; in TEST() local
732 const int in_cols = 7; in TEST() local
794 const int in_cols = 5; in TEST() local
846 const int in_cols = 5; in TEST() local
900 const int in_cols = 7; in TEST() local
966 const int in_cols = 7; in TEST() local
1032 const int in_cols = 5; in TEST() local
1087 const int in_cols = 5; in TEST() local
1143 const int in_cols = 5; in TEST() local
1213 const int in_cols = 5; in TEST() local
|
| D | avgpooling_op.cc | 310 const int64_t in_cols = output_shape.dim_size(2); in Compute() local
575 const int64 in_cols = output_shape.dim_size(2); in Compute() local
|
| D | pooling_ops_common.h | 235 const int32_t in_cols = params.tensor_in_cols; in SpatialMaxPool() local
|
| D | fractional_avg_pool_op.cc | 298 const int64_t in_cols = orig_input_tensor_shape_flat(2); in Compute() local
|
| D | conv_ops_3d.cc | 243 int64_t in_cols = GetTensorDim(input, data_format, '2'); in launch() local
|
| D | maxpooling_op.cc | 115 const int32_t in_cols = params.tensor_in_cols; in SpatialMaxPoolWithArgMaxHelper() local
617 const int32_t in_cols = params.tensor_in_cols; in SpatialMaxPoolGradGrad() local
|
| D | pooling_ops_3d.cc | 600 const int32_t in_cols = params.tensor_in_cols; in launch() local
|
| D | conv_ops.cc | 823 int64_t in_cols = GetTensorDim(input, data_format, 'W'); in operator ()() local
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | extract_volume_patches_op.cc | 81 const int in_cols = input.dim_size(3); in Compute() local
|
| D | extract_image_patches_op.cc | 74 const int in_cols = input.dim_size(2); in Compute() local
|
| /external/pytorch/aten/src/ATen/native/cpu/ |
| D | DepthwiseConvKernel.cpp | 27 int64_t in_cols; member
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 2675 auto in_cols = c->Value(in_cols_dim); in __anon847f0b683902() local
2781 auto in_cols = c->Value(in_cols_dim); in __anon847f0b683a02() local
|
| D | nn_ops.cc | 1065 auto in_cols = c->Value(in_cols_dim); in __anoncb4947e51702() local
|