| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dynamic_window_utils.cc | 115 result.padding_before = Zero<int32>(comp).get(); in GetWindowedOutputSize()
121 HloOp padding_before = padding_needed / ConstantR0<int32>(comp, 2); in GetWindowedOutputSize() local
122 result.padding_before = padding_before.get(); in GetWindowedOutputSize()
144 HloOp padding_before = effective_filter_size - One<int32>(comp) - in GetWindowedInputGradSize() local
145 HloOp(forward_dims.padding_before); in GetWindowedInputGradSize()
147 result.padding_before = padding_before.get(); in GetWindowedInputGradSize()
|
| D | dynamic_padder.cc | 746 absl::Span<HloInstruction*> padding_before, Window* input_window, in RewriteInputWithDynamicPadding() argument
761 if (padding_before[dim_index] == nullptr) { in RewriteInputWithDynamicPadding()
784 padding_before[dim_index])); in RewriteInputWithDynamicPadding()
817 std::vector<HloInstruction*> padding_before( in RewriteDynamicConvolutionInputGrad() local
842 padding_before[spatial_dim_index] = dynamic_window_dims.padding_before; in RewriteDynamicConvolutionInputGrad()
847 custom_call_conv, grad, zero, absl::MakeSpan(padding_before), &window, in RewriteDynamicConvolutionInputGrad()
885 std::vector<HloInstruction*> padding_before( in RewriteDynamicConvolutionForward() local
903 padding_before[spatial_dim_index] = dynamic_window_dims.padding_before; in RewriteDynamicConvolutionForward()
916 custom_call_conv, input, zero, absl::MakeSpan(padding_before), &window, in RewriteDynamicConvolutionForward()
946 std::vector<HloInstruction*> padding_before( in RewriteDynamicConvolutionKernelGrad() local
[all …]
|
| D | dynamic_window_utils.h | 31 HloInstruction* padding_before; member
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | kernel_shape_util.cc | 23 int64* padding_before, in GetWindowedOutputSizeVerboseV2() argument
38 *padding_before = *padding_after = 0; in GetWindowedOutputSizeVerboseV2()
41 *output_size = (input_size + *padding_before + *padding_after - in GetWindowedOutputSizeVerboseV2()
52 *padding_before = padding_needed / 2; in GetWindowedOutputSizeVerboseV2()
53 *padding_after = padding_needed - *padding_before; in GetWindowedOutputSizeVerboseV2()
68 int64* output_size, int64* padding_before, in GetWindowedOutputSizeVerbose() argument
73 padding_before, padding_after); in GetWindowedOutputSizeVerbose()
|
| D | kernel_shape_util.h | 116 int64* output_size, int64* padding_before,
124 int64* padding_before,
|
| D | common_shape_fns.h | 43 Padding padding_type, int64 padding_before, int64 padding_after,
|
| D | common_shape_fns.cc | 39 int64 stride, Padding padding_type, int64 padding_before, in GetWindowedOutputSizeFromDimsV2() argument
53 padding_before = padding_after = 0; in GetWindowedOutputSizeFromDimsV2()
57 c->Add(input_size, padding_before + padding_after, &input_size)); in GetWindowedOutputSizeFromDimsV2()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_shape_utils.cc | 57 Padding padding, int64 padding_before, int64 padding_after, int spatial_dim, in ConvBackpropExtractAndVerifyDimension() argument
67 &out_size, &padding_before, &padding_after)); in ConvBackpropExtractAndVerifyDimension()
80 dim->pad_before = effective_filter_size - 1 - padding_before; in ConvBackpropExtractAndVerifyDimension()
142 int64 padding_before = -1, padding_after = -1; in ConvBackpropComputeDimensionsV2() local
144 padding_before = explicit_paddings[2 * image_dim]; in ConvBackpropComputeDimensionsV2()
149 strides, padding, padding_before, padding_after, image_dim, i, in ConvBackpropComputeDimensionsV2()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_format.h | 500 char dimension, int64* padding_before, int64* padding_after) { in GetExplicitPaddingForDim() argument
503 *padding_before = explicit_paddings[2 * index]; in GetExplicitPaddingForDim()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 2648 int64 padding_before, padding_after; in __anondb9326b23902() local
2651 &padding_before, &padding_after)); in __anondb9326b23902()
2654 &padding_before, &padding_after)); in __anondb9326b23902()
2754 int64 padding_before, padding_after; in __anondb9326b23a02() local
2757 &padding_before, &padding_after)); in __anondb9326b23a02()
2760 &padding_before, &padding_after)); in __anondb9326b23a02()
2763 &padding_before, &padding_after)); in __anondb9326b23a02()
|
| D | nn_ops.cc | 1041 int64 padding_before, padding_after; in __anon4650176f1702() local
1044 &padding_before, &padding_after)); in __anon4650176f1702()
1047 &padding_before, &padding_after)); in __anon4650176f1702()
|
| /external/vixl/test/aarch32/ |
| D | test-assembler-aarch32.cc | 2254 const int padding_before = in TEST() local
2279 int space = AlignDown(padding_before, nop_size); in TEST()
|