Searched defs:concat_shape (Results 1 – 10 of 10) sorted by relevance
70 Shape concat_shape = ShapeUtil::MakeShape(F32, {10, 2 * kNumParams}); in TEST_F() local
128 …const TensorShape concat_shape = arm_compute::misc::shape_calculator::calculate_concatenate_shape(… in configure() local504 …const TensorShape concat_shape = arm_compute::misc::shape_calculator::calculate_concatenate_… in validate() local
42 const RuntimeShape& concat_shape, float* concat, in GruCell()
62 const RuntimeShape concat_shape = GetTensorShape(concat); in GruImpl() local
470 …const TensorShape concat_shape = arm_compute::misc::shape_calculator::calculate_concatenate_… in validate() local
164 Shape concat_shape; member
932 auto concat_shape = hlo->shape(); in ExchangeHalo() local
1914 const Shape concat_shape = ShapeUtil::MakeShape(F32, {200}); in TEST_F() local
2279 Shape concat_shape = in TEST_F() local2371 Shape concat_shape = ShapeUtil::MakeShape(F32, {50, 90}); in TEST_F() local
2221 Shape concat_shape({2, num_diags, max_diag_len}); in matchAndRewrite() local