Home
last modified time | relevance | path

Searched refs:after_padding (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dpad.cc84 int after_padding = *paddings_data++; in ResizeOutputTensor() local
86 TF_LITE_ENSURE_MSG(context, (before_padding >= 0 && after_padding >= 0), in ResizeOutputTensor()
90 (input_size->data[idx] + before_padding + after_padding); in ResizeOutputTensor()
137 std::vector<int> after_padding; in Eval() local
146 after_padding.push_back(paddings_data[idx * 2 + 1]); in Eval()
151 TF_LITE_ENSURE(context, after_padding.size() <= 4); \ in Eval()
154 op_params.right_padding_count = after_padding.size(); \ in Eval()
157 op_params.right_padding[i] = after_padding[op_context.dims - 1 - i]; \ in Eval()
/external/tensorflow/tensorflow/cc/gradients/
Darray_grad.cc480 Input after_padding = Reshape(scope, after_padding_sizes, padding_shape); in SliceGrad() local
485 Concat(scope, {before_padding, after_padding}, Const(scope, 1)); in SliceGrad()