Home
last modified time | relevance | path

Searched defs:in_cols (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dpad_test.cc344 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/
Ddepthwise_conv_op.h29 int in_cols; member
Ddeep_conv2d.h71 int in_cols; member
Dfused_batch_norm_op.cc124 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
Deigen_spatial_convolutions_test.cc671 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
Davgpooling_op.cc299 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
Dpooling_ops_common.h237 const int32 in_cols = params.tensor_in_cols; in SpatialMaxPool() local
Dfractional_avg_pool_op.cc255 const int64 in_cols = orig_input_tensor_shape_flat(2); in Compute() local
Dconv_ops_3d.cc225 int64 in_cols = GetTensorDim(input, data_format, '2'); in launch() local
Dmaxpooling_op.cc113 const int32 in_cols = params.tensor_in_cols; in SpatialMaxPoolWithArgMaxHelper() local
593 const int32 in_cols = params.tensor_in_cols; in SpatialMaxPoolGradGrad() local
Dpooling_ops_3d.cc577 const int32 in_cols = params.tensor_in_cols; in launch() local
Dconv_ops.cc668 int64 in_cols = GetTensorDim(input, data_format, 'W'); in operator ()() local
/external/tensorflow/tensorflow/core/kernels/image/
Dextract_volume_patches_op.cc81 const int in_cols = input.dim_size(3); in Compute() local
Dextract_image_patches_op.cc74 const int in_cols = input.dim_size(2); in Compute() local
/external/tensorflow/tensorflow/core/ops/
Dnn_ops.cc1031 auto in_cols = c->Value(in_cols_dim); in __anonad10f3361702() local
Darray_ops.cc2642 auto in_cols = c->Value(in_cols_dim); in __anon42d741193902() local
2748 auto in_cols = c->Value(in_cols_dim); in __anon42d741193a02() local