Home
last modified time | relevance | path

Searched defs:tensor_slice (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/uniform_quant_ops/
H A Dmath_utils_test.cc38 auto tensor_slice = tensor.tensor<float, 3>().slice(start_indices, sizes); in TEST() local
63 auto tensor_slice = tensor.tensor<qint8, 3>().slice(start_indices, sizes); in TEST() local
H A Dmath_utils.cc61 auto tensor_slice = in AsymmetricQuantize() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dload_and_remap_matrix_op.cc212 for (const TensorSlice& tensor_slice : tensor_slices) { in Compute() local