Searched defs:in_cols (Results 1 – 17 of 17) sorted by relevance
344 constexpr int64 in_cols = 35; in XLA_TEST_P() local366 constexpr int64 in_cols = 129; in XLA_TEST_P() local393 constexpr int64 in_cols = 129; in XLA_TEST_P() local421 constexpr int64 in_cols = 11; in XLA_TEST_P() local
29 int in_cols; member
71 int in_cols; member
71 const int in_cols = input.dim_size(2); in Compute() local
78 const int in_cols = input.dim_size(3); in Compute() local
671 const int in_cols = 7; in TEST() local732 const int in_cols = 7; in TEST() local794 const int in_cols = 5; in TEST() local846 const int in_cols = 5; in TEST() local900 const int in_cols = 7; in TEST() local966 const int in_cols = 7; in TEST() local1032 const int in_cols = 5; in TEST() local1087 const int in_cols = 5; in TEST() local1143 const int in_cols = 5; in TEST() local1213 const int in_cols = 5; in TEST() local
265 const int64 in_cols = output_shape.dim_size(2); in Compute() local506 const int64 in_cols = output_shape.dim_size(2); in Compute() local
125 const int in_rows, const int in_cols, const int out_planes, in MaxPool3DSYCL()
199 const int32 in_cols = params.tensor_in_cols; in SpatialMaxPool() local
255 const int64 in_cols = orig_input_tensor_shape_flat(2); in Compute() local
216 int64 in_cols = GetTensorDim(input, data_format, '2'); in launch() local
105 const int32 in_cols = params.tensor_in_cols; in SpatialMaxPoolWithArgMaxHelper() local585 const int32 in_cols = params.tensor_in_cols; in SpatialMaxPoolGradGrad() local
581 const int32 in_cols = params.tensor_in_cols; in launch() local
600 int64 in_cols = GetTensorDim(input, data_format, 'W'); in operator ()() local
2483 auto in_cols = c->Value(in_cols_dim); in __anon7c94107b3902() local2589 auto in_cols = c->Value(in_cols_dim); in __anon7c94107b3a02() local
937 auto in_cols = c->Value(in_cols_dim); in __anon3e672dd81d02() local
2611 int64 in_cols = input_desc.ndims() == 1 ? 1 : input_desc.width(); in ShouldIncludeWinogradNonfusedAlgo() local