Searched refs:padding_in (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/linalg/ |
| H A D | matrix_diag_op.cc | 92 const Tensor& padding_in = context->input(2); in Compute() local
93 OP_REQUIRES(context, padding_in.NumElements() == 1, in Compute()
95 padding_value = padding_in.flat<T>()(0); in Compute()
|
| /aosp_15_r20/external/ComputeLibrary/src/runtime/NEON/functions/ |
| H A D | NEFFTConvolutionLayer.cpp | 169 …const PaddingList padding_in = { { 0, kernel_size.x() + pad_valid.x() - 1 }, { 0, kernel_size.y() … in configure() local
171 _pad_input_func.configure(input_to_use, &_padded_input, padding_in); in configure()
|
| /aosp_15_r20/external/ComputeLibrary/src/runtime/CL/functions/ |
| H A D | CLFFTConvolutionLayer.cpp | 178 …const PaddingList padding_in = { { 0, kernel_size.x() + pad_valid.x() - 1 }, { 0, kernel_size.y() … in configure() local
180 _pad_input_func.configure(compile_context, input_to_use, &_padded_input, padding_in); in configure()
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/reference/ |
| H A D | DFT.cpp | 399 const PaddingList padding_in = { { 0, w.shape()[0] - 1 }, { 0, w.shape()[1] - 1 } }; in conv2d_dft() local
400 auto padded_src = pad_layer(src, padding_in); in conv2d_dft()
|