Searched refs:InferPadShape (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.h | 324 static StatusOr<Shape> InferPadShape(const Shape& operand_shape,
|
| D | hlo_creation_utils.cc | 94 ShapeInference::InferPadShape(operand->shape(), padding_value->shape(), in MakePadHlo()
|
| D | shape_inference_test.cc | 2114 auto inferred_status = ShapeInference::InferPadShape( in TEST_F()
2123 auto negative_dimension_size = ShapeInference::InferPadShape( in TEST_F()
|
| D | hlo_verifier.cc | 1348 return CheckShape(pad, ShapeInference::InferPadShape(pad->operand(0)->shape(), in HandlePad()
|
| D | hlo_evaluator_typed_visitor.h | 1358 ShapeInference::InferPadShape(
|
| D | algebraic_simplifier_test.cc | 4303 ShapeInference::InferPadShape(input->shape(), pad_value->shape(), in TEST_P()
4413 ShapeInference::InferPadShape(filter->shape(), pad_value->shape(), in TEST_P()
5733 ShapeInference::InferPadShape(input->shape(), in TEST_P()
|
| D | shape_inference.cc | 526 /* static */ StatusOr<Shape> ShapeInference::InferPadShape( in InferPadShape() function in xla::ShapeInference
|
| D | hlo_parser.cc | 2413 return ShapeInference::InferPadShape( in CreateInstruction()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner_util.cc | 501 auto padded_concat_shape = ShapeInference::InferPadShape( in PadFromPartialReplicateShape()
|
| D | spmd_partitioner.cc | 3435 ShapeInference::InferPadShape(sharded_pad->shape(), in HandlePad()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1154 Shape shape, ShapeInference::InferPadShape( in Pad()
|