Searched refs:InferConcatOpShape (Results 1 – 6 of 6) sorted by relevance
1354 auto inferred_status_1 = ShapeInference::InferConcatOpShape( in TEST_F()1360 auto inferred_status_2 = ShapeInference::InferConcatOpShape( in TEST_F()1366 auto inferred_status_3 = ShapeInference::InferConcatOpShape( in TEST_F()1377 ShapeInference::InferConcatOpShape({}, /*dimension=*/0); in TEST_F()1383 ShapeInference::InferConcatOpShape({&vector_32_}, /*dimension=*/-1); in TEST_F()1389 ShapeInference::InferConcatOpShape({&vector_32_}, /*dimension=*/1); in TEST_F()1395 auto inferred_status_error4 = ShapeInference::InferConcatOpShape( in TEST_F()1403 auto inferred_status_error5 = ShapeInference::InferConcatOpShape( in TEST_F()1409 auto inferred_status_error6 = ShapeInference::InferConcatOpShape( in TEST_F()
240 static StatusOr<Shape> InferConcatOpShape(
211 TF_ASSIGN_OR_RETURN(Shape concat_shape, ShapeInference::InferConcatOpShape( in MakeConcatHlo()
126 ShapeInference::InferConcatOpShape( in HandleConcatenate()
348 /* static */ StatusOr<Shape> ShapeInference::InferConcatOpShape( in InferConcatOpShape() function in xla::ShapeInference
861 TF_ASSIGN_OR_RETURN(Shape shape, ShapeInference::InferConcatOpShape( in ConcatInDim()