Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Ddepthwise_conv_op_gpu.h298 int shared_offset = data_idx;
306 const S* const tile_ptr = shared_offset + shared_data;
309 shared_offset += kBlockDepth;
314 shared_offset += in_increment;
591 int shared_offset = data_idx;
599 const S* const tile_ptr = shared_offset + shared_data;
602 ++shared_offset;
607 shared_offset += in_increment;
1270 int shared_offset = data_idx;
1274 const S* const tile_ptr = shared_offset + shared_data;
[all …]
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dconv_powervr.cc795 auto conv_core = [&](int shared_offset) { in GenerateConv() argument
810 int simd_id = (s * 4 + ch + shared_offset) / simd_size; in GenerateConv()
811 int thread_id = (s * 4 + ch + shared_offset) % simd_size; in GenerateConv()
845 std::to_string(s * 4 + ch + shared_offset); in GenerateConv()
876 std::to_string(s * 4 + i + shared_offset); in GenerateConv()