Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dconv_ops_using_gemm.cc331 const int64_t patch_index_start = chunk_index * patches_per_chunk; in operator ()() local
333 std::min(patch_index_start + patches_per_chunk, patch_count); in operator ()()
334 for (int64_t patch_index = patch_index_start; in operator ()()
409 const int how_many_patches = patch_index_end - patch_index_start; in operator ()()
416 T3* chunk_output_data = output_data + (patch_index_start * filter_count); in operator ()()
H A Dquantized_conv_ops.cc307 const int64_t patch_index_start = chunk_index * patches_per_chunk; in operator ()() local
309 std::min(patch_index_start + patches_per_chunk, patch_count); in operator ()()
310 for (int64_t patch_index = patch_index_start; in operator ()()
385 const int how_many_patches = patch_index_end - patch_index_start; in operator ()()
395 T3* chunk_output_data = output_data + (patch_index_start * filter_count); in operator ()()