Home
last modified time | relevance | path

Searched refs:src_coord (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_constant_strided_slice.cc53 std::vector<int> src_coord(num_input_axes); in StridedSlice() local
63 src_coord[axis] = start_index; in StridedSlice()
76 output_data[dst_offset] = input_buffer.data[Offset(input_shape, src_coord)]; in StridedSlice()
83 src_coord[axis] += stride; in StridedSlice()
89 if (!tflite::strided_slice::LoopCondition(src_coord[axis], stop, in StridedSlice()
93 src_coord[axis] = tflite::strided_slice::StartForAxis( in StridedSlice()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dpooling.cc62 std::string src_coord = src_coords[0]; in GetAveragePoolingKernelCode() local
64 src_coord += ", " + src_coords[i]; in GetAveragePoolingKernelCode()
124 c += " r += !outside ? args.src_tensor.Read<float>(" + src_coord + in GetAveragePoolingKernelCode()
128 c += " r += args.src_tensor.Read<float>(" + src_coord + ");\n"; in GetAveragePoolingKernelCode()
186 std::string src_coord = src_coords[0]; in GetMaxPoolingKernelCode() local
188 src_coord += ", " + src_coords[i]; in GetMaxPoolingKernelCode()
244 c += " FLT4 src = args.src_tensor.Read(" + src_coord + ");\n"; in GetMaxPoolingKernelCode()
Dconcat_xy.cc61 std::string src_coord = src_coords[0]; in GetConcatKernelCode() local
63 src_coord += ", " + src_coords[i]; in GetConcatKernelCode()
104 c += " result = args." + tensor_names[i] + ".Read(" + src_coord + ");\n"; in GetConcatKernelCode()
Dreduce.cc310 const std::string src_coord = "SRC_" + axis_to_coord[axis]; in GetReduceKernelCode() local
311 src_coords[axis] = src_coord; in GetReduceKernelCode()
312 c += " for (int " + src_coord + " = " + first + "; " + src_coord + in GetReduceKernelCode()
313 " < args.src_tensor." + axis_to_selector[axis] + "; " + src_coord + in GetReduceKernelCode()
/external/mesa3d/src/gallium/drivers/freedreno/a2xx/
Dir2_nir.c700 struct ir2_src src_coord = make_src_noconst(ctx, *coord); in emit_tex() local
712 instr->src[0] = src_coord; in emit_tex()
714 instr->src[1] = src_coord; in emit_tex()
728 src_coord = ir2_src(reg_idx, 0, IR2_SRC_REG); in emit_tex()
733 instr->src[0] = src_coord; in emit_tex()
/external/deqp/external/openglcts/modules/gl/
Dgl4cCopyImageTests.cpp3289 GLuint src_coord[3] = { 0, 0, 0 }; in prepareTestCases() local
3307 src_coord[1] = 1; in prepareTestCases()
3314 src_coord[1] = src_dim_diff / 2; in prepareTestCases()
3315 src_coord[2] = src_dim_diff; in prepareTestCases()
3349 test_case.m_src_x = src_coord[src_x]; in prepareTestCases()
3350 test_case.m_src_y = src_coord[src_y]; in prepareTestCases()
3360 test_case.m_src_x = src_coord[0]; in prepareTestCases()
3361 test_case.m_src_y = src_coord[0]; in prepareTestCases()
/external/mesa3d/src/amd/vulkan/
Dradv_meta_bufimage.c746 nir_ssa_def *src_coord = nir_iadd(&b, global_id, &src_offset->dest.ssa); in build_nir_itoi_compute_shader() local
755 tex->src[0].src = nir_src_for_ssa(nir_channels(&b, src_coord, is_3d ? 0x7 : 0x3)); in build_nir_itoi_compute_shader()
985 nir_ssa_def *src_coord = in build_nir_itoi_r32g32b32_compute_shader() local
995 tex->src[0].src = nir_src_for_ssa(nir_channels(&b, src_coord, 1)); in build_nir_itoi_r32g32b32_compute_shader()
/external/llvm-project/clang/lib/Headers/
Dopencl-c.h16517 ushort2 src_coord, uchar partition_mask, uchar sad_adjustment);
16547 short2 ref_offset, ushort2 src_coord, ushort2 ref_window_size,
16638 ushort2 src_coord, ulong motion_vectors, uchar major_shapes,
16643 ushort2 src_coord, ulong motion_vectors, uchar major_shapes,
16676 ushort2 src_coord);