Searched refs:padding_shape (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | xla_conv_op.cc | 46 const TensorShape padding_shape = context->InputShape("padding"); in Compile() local
61 TensorShapeUtils::IsMatrix(padding_shape) && in Compile()
62 padding_shape.dim_size(1) == 2, in Compile()
65 padding_shape.DebugString())); in Compile()
69 std::vector<std::pair<int64, int64>> padding(padding_shape.dim_size(0)); in Compile()
|
| D | reduce_window_op.cc | 103 const TensorShape padding_shape = context->InputShape("padding"); in Compile() local
105 TensorShapeUtils::IsMatrix(padding_shape) && in Compile()
106 padding_shape.dim_size(1) == 2, in Compile()
109 padding_shape.DebugString())); in Compile()
113 std::vector<std::pair<int64, int64>> padding(padding_shape.dim_size(0)); in Compile()
|
| D | xla_select_and_scatter_op.cc | 110 const TensorShape padding_shape = context->InputShape("padding"); in Compile() local
112 TensorShapeUtils::IsMatrix(padding_shape) && in Compile()
113 padding_shape.dim_size(1) == 2, in Compile()
116 padding_shape.DebugString())); in Compile()
120 std::vector<std::pair<int64, int64>> padding(padding_shape.dim_size(0)); in Compile()
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | array_grad.cc | 467 auto padding_shape = Stack(scope, {input_rank, 1}); in SliceGrad() local
471 Input before_padding = Reshape(scope, begin, padding_shape); in SliceGrad()
480 Input after_padding = Reshape(scope, after_padding_sizes, padding_shape); in SliceGrad()
|
| /external/tensorflow/tensorflow/contrib/training/python/training/ |
| D | sequence_queueing_state_saver.py | 1614 padding_shape = array_ops.concat(
1619 paddings = array_ops.fill(dims=padding_shape, value=dummy)
|