Searched refs:InferConcatOpShape (Results 1 – 7 of 7) sorted by relevance
1993 auto inferred_status = ShapeInference::InferConcatOpShape( in TEST_F()2003 auto inferred_status_1 = ShapeInference::InferConcatOpShape( in TEST_F()2009 auto inferred_status_2 = ShapeInference::InferConcatOpShape( in TEST_F()2015 auto inferred_status_3 = ShapeInference::InferConcatOpShape( in TEST_F()2026 ShapeInference::InferConcatOpShape({}, /*dimension=*/0); in TEST_F()2032 ShapeInference::InferConcatOpShape({&vector_32_}, /*dimension=*/-1); in TEST_F()2038 ShapeInference::InferConcatOpShape({&vector_32_}, /*dimension=*/1); in TEST_F()2044 auto inferred_status_error4 = ShapeInference::InferConcatOpShape( in TEST_F()2052 auto inferred_status_error5 = ShapeInference::InferConcatOpShape( in TEST_F()2058 auto inferred_status_error6 = ShapeInference::InferConcatOpShape( in TEST_F()
269 static StatusOr<Shape> InferConcatOpShape(
246 TF_ASSIGN_OR_RETURN(Shape concat_shape, ShapeInference::InferConcatOpShape( in MakeConcatHlo()
120 ShapeInference::InferConcatOpShape( in HandleConcatenate()
382 /* static */ StatusOr<Shape> ShapeInference::InferConcatOpShape( in InferConcatOpShape() function in xla::ShapeInference
1775 return ShapeInference::InferConcatOpShape(arg_shapes, in ParseInstructionRhs()
1064 TF_ASSIGN_OR_RETURN(Shape shape, ShapeInference::InferConcatOpShape( in ConcatInDim()