Searched refs:InferPadShape (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.h | 293 static StatusOr<Shape> InferPadShape(const Shape& operand_shape,
|
| D | hlo_creation_utils.cc | 75 ShapeInference::InferPadShape(operand->shape(), padding_value->shape(), in MakePadHlo()
|
| D | shape_inference_test.cc | 2082 auto inferred_status = ShapeInference::InferPadShape( in TEST_F()
2091 auto negative_dimension_size = ShapeInference::InferPadShape( in TEST_F()
|
| D | hlo_verifier.cc | 978 return CheckShape(pad, ShapeInference::InferPadShape(pad->operand(0)->shape(), in HandlePad()
|
| D | algebraic_simplifier_test.cc | 3867 ShapeInference::InferPadShape(input->shape(), pad_value->shape(), in TEST_P()
3977 ShapeInference::InferPadShape(filter->shape(), pad_value->shape(), in TEST_P()
5093 ShapeInference::InferPadShape(input->shape(), in TEST_P()
|
| D | shape_inference.cc | 529 /* static */ StatusOr<Shape> ShapeInference::InferPadShape( in InferPadShape() function in xla::ShapeInference
|
| D | hlo_evaluator_typed_visitor.h | 1541 ShapeInference::InferPadShape(
|
| D | hlo_parser.cc | 2061 return ShapeInference::InferPadShape( in ParseInstructionRhs()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner_util.cc | 585 auto padded_concat_shape = ShapeInference::InferPadShape( in PadFromPartialReplicateShape()
|
| D | spmd_partitioner.cc | 2666 ShapeInference::InferPadShape(sharded_pad->shape(), in HandlePad()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1087 Shape shape, ShapeInference::InferPadShape( in Pad()
|