Searched refs:InferReverseShape (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.h | 190 static StatusOr<Shape> InferReverseShape(const Shape& operand_shape,
|
| D | shape_inference_test.cc | 2101 auto inferred_status = ShapeInference::InferReverseShape(input_shape, {0, 1}); in TEST_F()
2111 ShapeInference::InferReverseShape(input_shape, {0, 2}); in TEST_F()
2117 ShapeInference::InferReverseShape(input_shape, {0, -1}); in TEST_F()
2123 ShapeInference::InferReverseShape(input_shape, {0, 0}); in TEST_F()
2130 ShapeInference::InferReverseShape(tuple_shape, {0}); in TEST_F()
|
| D | hlo_creation_utils.cc | 375 TF_ASSIGN_OR_RETURN(Shape reverse_shape, ShapeInference::InferReverseShape( in MakeReverseHlo()
|
| D | hlo_verifier.cc | 538 reverse, ShapeInference::InferReverseShape(reverse->operand(0)->shape(), in HandleReverse()
|
| D | shape_inference.cc | 2712 /*static */ StatusOr<Shape> ShapeInference::InferReverseShape( in InferReverseShape() function in xla::ShapeInference
|
| D | hlo_evaluator_typed_visitor.h | 1062 ShapeInference::InferReverseShape(operand->shape(),
|
| D | hlo_parser.cc | 1863 return ShapeInference::InferReverseShape(operands[0]->shape(), in ParseInstructionRhs()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 2043 TF_ASSIGN_OR_RETURN(Shape shape, ShapeInference::InferReverseShape( in Rev()
|