Searched defs:in_rows (Results 1 – 17 of 17) sorted by relevance
343 constexpr int64 in_rows = 35; in XLA_TEST_P() local365 constexpr int64 in_rows = 129; in XLA_TEST_P() local392 constexpr int64 in_rows = 129; in XLA_TEST_P() local420 constexpr int64 in_rows = 8; in XLA_TEST_P() local
28 int in_rows; member
70 int in_rows; member
70 const int in_rows = input.dim_size(1); in Compute() local
77 const int in_rows = input.dim_size(2); in Compute() local
670 const int in_rows = 8; in TEST() local731 const int in_rows = 8; in TEST() local793 const int in_rows = 5; in TEST() local845 const int in_rows = 5; in TEST() local899 const int in_rows = 8; in TEST() local965 const int in_rows = 8; in TEST() local1031 const int in_rows = 7; in TEST() local1086 const int in_rows = 7; in TEST() local1142 const int in_rows = 7; in TEST() local1212 const int in_rows = 7; in TEST() local
264 const int64 in_rows = output_shape.dim_size(1); in Compute() local505 const int64 in_rows = output_shape.dim_size(1); in Compute() local
125 const int in_rows, const int in_cols, const int out_planes, in MaxPool3DSYCL()
198 const int32 in_rows = params.tensor_in_rows; in SpatialMaxPool() local
254 const int64 in_rows = orig_input_tensor_shape_flat(1); in Compute() local
215 int64 in_rows = GetTensorDim(input, data_format, '1'); in launch() local
104 const int32 in_rows = params.tensor_in_rows; in SpatialMaxPoolWithArgMaxHelper() local584 const int32 in_rows = params.tensor_in_rows; in SpatialMaxPoolGradGrad() local
580 const int32 in_rows = params.tensor_in_rows; in launch() local
599 int64 in_rows = GetTensorDim(input, data_format, 'H'); in operator ()() local
2482 auto in_rows = c->Value(in_rows_dim); in __anon7c94107b3902() local2588 auto in_rows = c->Value(in_rows_dim); in __anon7c94107b3a02() local
936 auto in_rows = c->Value(in_rows_dim); in __anon3e672dd81d02() local
2610 int64 in_rows = input_desc.height(); in ShouldIncludeWinogradNonfusedAlgo() local