Searched refs:InferGatherShape (Results 1 – 7 of 7) sorted by relevance
2427 ShapeInference::InferGatherShape( in TEST_F()2442 ShapeInference::InferGatherShape( in TEST_F()2457 ShapeInference::InferGatherShape( in TEST_F()2473 ShapeInference::InferGatherShape( in TEST_F()2490 ShapeInference::InferGatherShape( in TEST_F()2507 ShapeInference::InferGatherShape( in TEST_F()2524 ShapeInference::InferGatherShape( in TEST_F()2542 ShapeInference::InferGatherShape( in TEST_F()2560 ShapeInference::InferGatherShape( in TEST_F()2578 ShapeInference::InferGatherShape( in TEST_F()[all …]
313 static StatusOr<Shape> InferGatherShape(
1134 ShapeInference::InferGatherShape( in HandleGather()
3302 /*static*/ StatusOr<Shape> ShapeInference::InferGatherShape( in InferGatherShape() function in xla::ShapeInference
2473 return ShapeInference::InferGatherShape( in ParseInstructionRhs()
148 ShapeInference::InferGatherShape(replicated_operand.hlo()->shape(), in PartitionIndexOnlyPartition()
2278 TF_ASSIGN_OR_RETURN(Shape shape, ShapeInference::InferGatherShape( in Gather()