Searched refs:padding_value_shape (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | xla_pad_op.cc | 34 const TensorShape padding_value_shape = in Compile() local
47 OP_REQUIRES(context, TensorShapeUtils::IsScalar(padding_value_shape), in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.h | 265 const Shape& padding_value_shape,
|
| D | shape_inference.cc | 485 const Shape& operand_shape, const Shape& padding_value_shape, in InferPadShape() argument
491 if (!ShapeUtil::IsScalar(padding_value_shape)) { in InferPadShape()
503 padding_value_shape)) { in InferPadShape()
515 if (!padding_value_shape.is_static()) { in InferPadShape()
540 ShapeUtil::HigherPrecisionElementType(operand_shape, padding_value_shape), in InferPadShape()
|
| D | shape_inference_test.cc | 1420 Shape padding_value_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
1434 input_shape, padding_value_shape, padding_config); in TEST_F()
1443 input_shape, padding_value_shape, padding_config); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 877 TF_ASSIGN_OR_RETURN(const Shape& padding_value_shape, in Pad()
881 operand_shape, padding_value_shape, padding_config)); in Pad()
|