Searched defs:concat_shape (Results 1 – 9 of 9) sorted by relevance
70 Shape concat_shape = ShapeUtil::MakeShape(F32, {10, 2 * kNumParams}); in TEST_F() local
147 auto concat_shape = concat.shape(); in Compute() local
42 const RuntimeShape& concat_shape, float* concat, in GruCell()
60 const RuntimeShape concat_shape = GetTensorShape(concat); in GruImpl() local
164 Shape concat_shape; member
886 auto concat_shape = hlo->shape(); in ExchangeHalo() local
1875 const Shape concat_shape = ShapeUtil::MakeShape(F32, {200}); in TEST_F() local
2246 Shape concat_shape = in TEST_F() local2338 Shape concat_shape = ShapeUtil::MakeShape(F32, {50, 90}); in TEST_F() local
1614 Shape concat_shape({2, num_diags, max_diag_len}); in matchAndRewrite() local