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 | 102 const TensorShape padding_shape = context->InputShape("padding"); in Compile() local
104 TensorShapeUtils::IsMatrix(padding_shape) && in Compile()
105 padding_shape.dim_size(1) == 2, in Compile()
108 padding_shape.DebugString())); in Compile()
112 std::vector<std::pair<int64, int64>> padding(padding_shape.dim_size(0)); in Compile()
|
| D | xla_select_and_scatter_op.cc | 109 const TensorShape padding_shape = context->InputShape("padding"); in Compile() local
111 TensorShapeUtils::IsMatrix(padding_shape) && in Compile()
112 padding_shape.dim_size(1) == 2, in Compile()
115 padding_shape.DebugString())); in Compile()
119 std::vector<std::pair<int64, int64>> padding(padding_shape.dim_size(0)); in Compile()
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | array_grad.cc | 476 auto padding_shape = Stack(scope, {input_rank, 1}); in SliceGrad() local
480 Input before_padding = Reshape(scope, begin, padding_shape); in SliceGrad()
489 Input after_padding = Reshape(scope, after_padding_sizes, padding_shape); in SliceGrad()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | model_builder.cc | 2248 auto padding_shape = padding.shape; in Parse() local
2250 if (padding_shape.h != 2 && padding_shape.w != 2) { in Parse()
|