Searched refs:pad_shape (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | pad_op.cc | 34 const TensorShape pad_shape = ctx->InputShape("paddings"); in Compile() local
38 TensorShapeUtils::IsMatrix(pad_shape) && pad_shape.dim_size(1) == 2, in Compile()
40 pad_shape.DebugString())); in Compile()
42 ctx, dims == pad_shape.dim_size(0), in Compile()
45 pad_shape.DebugString(), " ", input_shape.DebugString())); in Compile()
|
| D | mirror_pad_op.cc | 66 const TensorShape pad_shape = ctx->InputShape("paddings"); in Compile() local
78 TensorShapeUtils::IsMatrix(pad_shape) && pad_shape.dim_size(1) == 2, in Compile()
80 pad_shape.DebugString())); in Compile()
82 ctx, dims == pad_shape.dim_size(0), in Compile()
85 pad_shape.DebugString(), " ", input_shape.DebugString())); in Compile()
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_batch_gather_with_default_op.py | 116 pad_shape = _get_pad_shape(params, indices)
119 default_value, pad_shape)
122 pad_shape = array_ops.concat([
128 pad = array_ops.broadcast_to(default_value, pad_shape)
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_creation_utils.cc | 63 Shape pad_shape, in MakePadHlo()
67 pad_shape, operand, padding_value, padding_config)); in MakePadHlo()
|
| D | algebraic_simplifier_test.cc | 4235 const Shape pad_shape, in TEST_P()
4239 pad_shape, input, in TEST_P()
|