Searched refs:InferTransposeShape (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.h | 264 static StatusOr<Shape> InferTransposeShape(
|
| D | hlo_creation_utils.cc | 116 ShapeInference::InferTransposeShape(operand->shape(), dimensions)); in MakeTransposeHlo()
|
| D | shape_inference_test.cc | 2172 ShapeInference::InferTransposeShape(a_shape, {1, 2, 3, 0}); in TEST_F()
2182 ShapeInference::InferTransposeShape(a_shape, {0}); in TEST_F()
|
| D | hlo_verifier.cc | 740 transpose, ShapeInference::InferTransposeShape( in HandleTranspose()
|
| D | shape_inference.cc | 3094 /* static */ StatusOr<Shape> ShapeInference::InferTransposeShape( in InferTransposeShape() function in xla::ShapeInference
|
| D | hlo_parser.cc | 1964 return ShapeInference::InferTransposeShape(operands[0]->shape(), in ParseInstructionRhs()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.cc | 1203 ShapeInference::InferTransposeShape(all_to_all->shape(), permutation) in ReshardWithAllToAll()
3585 ShapeInference::InferTransposeShape(result->shape(), xpose_permutation) in AllGatherShardsInternal()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 2024 TF_ASSIGN_OR_RETURN(Shape shape, ShapeInference::InferTransposeShape( in Transpose()
|