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 | 294 const Shape& padding_value_shape,
|
| D | shape_inference.cc | 530 const Shape& operand_shape, const Shape& padding_value_shape, in InferPadShape() argument
536 if (!ShapeUtil::IsScalar(padding_value_shape)) { in InferPadShape()
548 padding_value_shape)) { in InferPadShape()
560 if (!padding_value_shape.is_static()) { in InferPadShape()
580 ShapeUtil::HigherPrecisionElementType(operand_shape, padding_value_shape), in InferPadShape()
|
| D | shape_inference_test.cc | 2069 Shape padding_value_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
2083 input_shape, padding_value_shape, padding_config); in TEST_F()
2092 input_shape, padding_value_shape, padding_config); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1084 TF_ASSIGN_OR_RETURN(const Shape* padding_value_shape, in Pad()
1088 *operand_shape, *padding_value_shape, padding_config)); in Pad()
|