Home
last modified time | relevance | path

Searched defs:dst_index (Results 1 – 25 of 41) sorted by relevance

12

/external/pytorch/aten/src/ATen/native/cuda/
DUpSample.cuh117 int dst_index, in area_pixel_compute_source_index()
135 int dst_index, in nearest_neighbor_compute_source_index()
149 int dst_index, in nearest_neighbor_exact_compute_source_index()
162 int dst_index, in nearest_neighbor_bw_compute_source_index()
175 int dst_index, in nearest_neighbor_exact_bw_compute_source_index()
DUpSampleNearest2d.cu70 size_t dst_index = (nc_iter * height2 + h2) * width2 + w2; in upsample_nearest2d_out_frame() local
/external/mesa3d/src/gallium/auxiliary/tgsi/
Dtgsi_transform.h293 unsigned dst_index, in tgsi_transform_op1_inst()
317 unsigned dst_index, in tgsi_transform_op2_inst()
346 unsigned dst_index, in tgsi_transform_op1_swz_inst()
387 unsigned dst_index, in tgsi_transform_op2_swz_inst()
438 unsigned dst_index, in tgsi_transform_op3_swz_inst()
520 unsigned dst_index, in tgsi_transform_tex_inst()
/external/ComputeLibrary/tests/validation/reference/
DTranspose.cpp52 const size_t dst_index = coord2index(dst.shape(), dst_coord); in transpose() local
DPooling3dLayer.cpp162 … int dst_index = batch_offset_dst + depth_offset_dst + height_offset_dst + w * num_channels + c; in pooling_3d_layer_internal() local
/external/ruy/ruy/profiler/
Dtest_instrumented_library.cc35 int dst_index = 0; in MergeSortRecurse() local
/external/v4l-utils/lib/libv4lconvert/
Dsn9c2028-decomp.c127 long dst_index = 0; in v4lconvert_decode_sn9c2028() local
/external/pytorch/aten/src/ATen/native/
DUpSample.h286 int64_t dst_index, in area_pixel_compute_source_index()
313 int64_t dst_index, in nearest_neighbor_compute_source_index()
324 int64_t dst_index, in nearest_neighbor_exact_compute_source_index()
/external/pytorch/aten/src/ATen/native/vulkan/ops/
DMm.h42 int64_t dst_index = (src_h / 2) * dst_kw_sz + (src_w / 2); in stage_pack_weights() local
DMm.cpp249 int64_t dst_index = (src_h / 2) * dst_kw_sz + (src_w / 2); in pack_biases_quantized_weights() local
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_relu_op.cc443 const size_t dst_index = 0; // index of dst output tensor in Compute() local
775 const size_t dst_index = 0; // index of dst output tensor in Compute_Scalar() local
842 const size_t dst_index = 0; // index of dst output tensor in Compute_Scalar() local
923 const size_t dst_index = 0; // index of dst output tensor in Compute_Scalar() local
1008 const size_t dst_index = 0; // index of dst output tensor in Compute_Scalar() local
1084 const size_t dst_index = 0; // index of dst output tensor in Compute_Scalar() local
/external/mesa3d/src/imagination/rogue/
Drogue_compile.c111 unsigned dst_index = load_const->def.index; in trans_nir_load_const() local
608 unsigned dst_index = alu->def.index; in trans_nir_alu_vecN() local
621 unsigned dst_index = alu->def.index; in trans_nir_alu_iadd64() local
Drogue.h304 unsigned dst_index; member
994 unsigned dst_index; member
1297 unsigned dst_index, in rogue_set_alu_dst_mod()
1304 unsigned dst_index, in rogue_alu_dst_mod_is_set()
1784 unsigned dst_index; member
2213 unsigned dst_index) in rogue_link_instr_write_reg()
2231 unsigned dst_index) in rogue_link_instr_write_regarray()
2302 unsigned dst_index = write->dst_index; in rogue_dst_reg_replace() local
2400 unsigned dst_index = write->dst_index; in rogue_dst_regarray_replace() local
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dfully_connected.h78 int dst_index = block_x * padded_dst_channels * 4 + block_y * 16 + in RearrangeFCWeightsToIOO4I4() local
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/
Dfc_fc_add.h57 int dst_index = block_x * padded_dst_channels * 4 + block_y * 16 + in RearrangeFCWeightsToIOO4I4() local
/external/mesa3d/src/virtio/vulkan/
Dvn_query_pool.c184 uint32_t dst_index = 0; in vn_get_query_pool_feedback() local
/external/tensorflow/tensorflow/core/graph/
Dgraph.cc686 int dst_index) { in UpdateEdge()
711 int dst_index = 0; in AddWhileInputHack() local
/external/tensorflow/tensorflow/compiler/xla/
Dshape_tree.h269 const ShapeIndex& dst_index) { in CopySubtreeFrom()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dexport_graphdef.cc268 unsigned dst_index) { in AddEdgeBetweenNodes()
/external/mesa3d/src/intel/compiler/
Dbrw_nir_lower_cooperative_matrix.c694 const nir_src dst_index = intrin->src[3]; in lower_cmat_instr() local
/external/mesa3d/src/gallium/drivers/radeonsi/
Dsi_compute_blit.c808 unsigned dst_index = is_clear ? 0 : 1; in si_compute_blit() local
/external/mesa3d/src/gallium/drivers/r300/compiler/
Dradeon_dataflow.c600 rc_register_file dst_file, unsigned int dst_index, in get_readers_for_single_write()
/external/mesa3d/src/gallium/drivers/llvmpipe/
Dlp_texture_handle.c835 uint32_t dst_index = matrix->texture_count; in llvmpipe_register_texture() local
/external/mesa3d/src/gallium/drivers/nouveau/nv50/
Dnv50_state.c1115 unsigned dst_index = i; in nv50_set_vertex_buffers() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator.cc1951 bool within_src_bounds) { in GenerateIndices()
1999 bool within_src_bounds) { in CopyDataFromInput()
2050 bool within_src_bounds) { in CopyDataToOutput()

12