Home
last modified time | relevance | path

Searched defs:in_cols (Results 1 – 17 of 17) 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
Dextract_image_patches_op.cc71 const int in_cols = input.dim_size(2); in Compute() local
Dextract_volume_patches_op.cc78 const int in_cols = input.dim_size(3); in Compute() 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.cc265 const int64 in_cols = output_shape.dim_size(2); in Compute() local
506 const int64 in_cols = output_shape.dim_size(2); in Compute() local
Dpooling_ops_3d_sycl.h125 const int in_rows, const int in_cols, const int out_planes, in MaxPool3DSYCL()
Dpooling_ops_common.h199 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.cc216 int64 in_cols = GetTensorDim(input, data_format, '2'); in launch() local
Dmaxpooling_op.cc105 const int32 in_cols = params.tensor_in_cols; in SpatialMaxPoolWithArgMaxHelper() local
585 const int32 in_cols = params.tensor_in_cols; in SpatialMaxPoolGradGrad() local
Dpooling_ops_3d.cc581 const int32 in_cols = params.tensor_in_cols; in launch() local
Dconv_ops.cc600 int64 in_cols = GetTensorDim(input, data_format, 'W'); in operator ()() local
/external/tensorflow/tensorflow/core/ops/
Darray_ops.cc2483 auto in_cols = c->Value(in_cols_dim); in __anon7c94107b3902() local
2589 auto in_cols = c->Value(in_cols_dim); in __anon7c94107b3a02() local
Dnn_ops.cc937 auto in_cols = c->Value(in_cols_dim); in __anon3e672dd81d02() local
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc2611 int64 in_cols = input_desc.ndims() == 1 ? 1 : input_desc.width(); in ShouldIncludeWinogradNonfusedAlgo() local