Home
last modified time | relevance | path

Searched defs:k_val (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/lhlo_gpu_to_tfrt_gpu/
Dgemm_pattern.cc137 auto k_val = lhs_matrix.transpose ? lhs_matrix.num_rows : lhs_matrix.num_cols; in CreateTfrtOps() local
/external/ComputeLibrary/src/core/NEON/kernels/
DNEDirectConvolutionLayerKernel.cpp211 …const auto k_val = reinterpret_cast<const float *>(k_ptr + p * kernel_stride_z + (id.z() + oz) * k… in convolve() local
291 …const auto k_val = reinterpret_cast<const T1 *>(k_ptr + 0 * kernel_stride_z + (id.z() + oz) * kern… in convolve() local
308 …const auto k_val = reinterpret_cast<const T1 *>(k_ptr + p * kernel_stride_z + (id.z() + oz) * kern… in convolve() local