/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | pad_test.cc | 344 constexpr int64 in_cols = 35; in XLA_TEST_P() local 366 constexpr int64 in_cols = 129; in XLA_TEST_P() local 393 constexpr int64 in_cols = 129; in XLA_TEST_P() local 421 constexpr int64 in_cols = 11; in XLA_TEST_P() 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 | 124 const int64 in_cols = GetTensorDim(x_input, tensor_format, 'W'); in operator ()() local 262 const int64 in_cols = GetTensorDim(x_input, tensor_format, 'W'); in operator ()() local 354 const int64 in_cols = GetTensorDim(x_input, tensor_format, 'W'); in operator ()() local 1274 int64 in_cols = GetTensorDim(x, tensor_format_, '2'); in ComputeWithReservedSpace() local 1435 int64 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 | 299 const int64 in_cols = output_shape.dim_size(2); in Compute() local 563 const int64 in_cols = output_shape.dim_size(2); in Compute() local
|
D | pooling_ops_common.h | 237 const int32 in_cols = params.tensor_in_cols; in SpatialMaxPool() local
|
D | fractional_avg_pool_op.cc | 255 const int64 in_cols = orig_input_tensor_shape_flat(2); in Compute() local
|
D | conv_ops_3d.cc | 225 int64 in_cols = GetTensorDim(input, data_format, '2'); in launch() local
|
D | maxpooling_op.cc | 113 const int32 in_cols = params.tensor_in_cols; in SpatialMaxPoolWithArgMaxHelper() local 593 const int32 in_cols = params.tensor_in_cols; in SpatialMaxPoolGradGrad() local
|
D | pooling_ops_3d.cc | 577 const int32 in_cols = params.tensor_in_cols; in launch() local
|
D | conv_ops.cc | 668 int64 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/tensorflow/tensorflow/core/ops/ |
D | nn_ops.cc | 1031 auto in_cols = c->Value(in_cols_dim); in __anonad10f3361702() local
|
D | array_ops.cc | 2642 auto in_cols = c->Value(in_cols_dim); in __anon42d741193902() local 2748 auto in_cols = c->Value(in_cols_dim); in __anon42d741193a02() local
|