Searched refs:src_coord (Results 1 – 8 of 8) sorted by relevance
/external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
D | resolve_constant_strided_slice.cc | 53 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/ |
D | pooling.cc | 62 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()
|
D | concat_xy.cc | 61 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()
|
D | reduce.cc | 310 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/ |
D | ir2_nir.c | 700 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/ |
D | gl4cCopyImageTests.cpp | 3289 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/ |
D | radv_meta_bufimage.c | 746 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/ |
D | opencl-c.h | 16517 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);
|