Searched refs:InferPadShape (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.h | 264 static StatusOr<Shape> InferPadShape(const Shape& operand_shape,
|
| D | hlo_creation_utils.cc | 64 ShapeInference::InferPadShape(operand->shape(), padding_value->shape(), in MakePadHlo()
|
| D | hlo_verifier.cc | 690 return CheckShape(pad, ShapeInference::InferPadShape(pad->operand(0)->shape(), in HandlePad()
|
| D | shape_inference_test.cc | 1433 auto inferred_status = ShapeInference::InferPadShape( in TEST_F()
1442 auto negative_dimension_size = ShapeInference::InferPadShape( in TEST_F()
|
| D | algebraic_simplifier_test.cc | 3107 ShapeInference::InferPadShape(input->shape(), pad_value->shape(), in TEST_P()
3215 ShapeInference::InferPadShape(filter->shape(), pad_value->shape(), in TEST_P()
4236 ShapeInference::InferPadShape(input->shape(), in TEST_P()
|
| D | shape_inference.cc | 484 /* static */ StatusOr<Shape> ShapeInference::InferPadShape( in InferPadShape() function in xla::ShapeInference
|
| D | hlo_evaluator_typed_visitor.h | 1397 ShapeInference::InferPadShape( in HandlePad()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 880 Shape shape, ShapeInference::InferPadShape( in Pad()
|