Home
last modified time | relevance | path

Searched defs:src_index (Results 1 – 25 of 65) sorted by relevance

123

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_relu_op.cc442 const size_t src_index = 0; // index of src input tensor in Compute() local
593 size_t src_index = GetSrcIndex(); in Compute() local
774 const size_t src_index = 0; // index of src input tensor in Compute_Scalar() local
806 const size_t src_index = 1; // index of src input tensor in Compute_Scalar() local
841 const size_t src_index = 0; // index of src input tensor in Compute_Scalar() local
877 const size_t src_index = 1; // index of src input tensor in Compute_Scalar() local
922 const size_t src_index = 0; // index of src input tensor in Compute_Scalar() local
968 const size_t src_index = GetSrcIndex(); in Compute_Scalar() local
1007 const size_t src_index = 0; // index of src input tensor in Compute_Scalar() local
1039 const size_t src_index = 1; // index of src input tensor in Compute_Scalar() local
[all …]
H A Dmkl_conv_ops.h290 virtual inline void GetFilterSizeInMklOrder(size_t src_index, in GetFilterSizeInMklOrder()
540 size_t src_index, size_t filter_index, const memory::dims& strides, in GetOutputAndPadSizeInMklOrder()
H A Dmkl_aggregate_ops.cc41 TensorShape GetTensorShape(OpKernelContext* ctx, size_t src_index) { in GetTensorShape()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DUpSample.cuh142 const int src_index = in nearest_neighbor_compute_source_index() local
154 const int src_index = in nearest_neighbor_exact_compute_source_index() local
167 const int src_index = in nearest_neighbor_bw_compute_source_index() local
178 const int src_index = in nearest_neighbor_exact_bw_compute_source_index() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dtuple_points_to_analysis.cc239 ShapeIndex src_index; in HandleGetTupleElement() local
371 const PointsToSet::BufferList& points_to) { in HandleAsyncDone()
418 const PointsToSet::BufferList& points_to) { in HandleCopyDone()
456 const PointsToSet::BufferList& points_to) { in HandleSend()
490 const PointsToSet::BufferList& points_to) { in HandleTuple()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dnode_def_builder.cc96 NodeDefBuilder& NodeDefBuilder::Input(StringPiece src_node, int src_index, in Input()
116 StringPiece src_node, int src_index, in SingleInput()
173 void NodeDefBuilder::AddInput(StringPiece src_node, int src_index) { in AddInput()
/aosp_15_r20/external/mesa3d/src/gallium/auxiliary/draw/
H A Ddraw_vertex.h77 unsigned src_index:8; /**< map to post-xform attribs */ member
117 int src_index) in draw_emit_vertex_attr()
/aosp_15_r20/external/mesa3d/src/gallium/drivers/r300/compiler/
H A Dradeon_optimize.c378 unsigned int src_index; in presub_helper() local
394 unsigned int src_index) in presub_replace_add()
515 unsigned int src_index) in presub_replace_inv()
532 unsigned int src_index) in presub_replace_bias()
/aosp_15_r20/external/mesa3d/src/imagination/rogue/
H A Drogue.h319 unsigned src_index; member
669 unsigned src_index; member
1006 unsigned src_index; member
1311 unsigned src_index, in rogue_set_alu_src_mod()
1318 unsigned src_index, in rogue_alu_src_mod_is_set()
2191 unsigned src_index, in rogue_link_imm_use()
2250 unsigned src_index) in rogue_link_instr_use_reg()
2268 unsigned src_index) in rogue_link_instr_use_regarray()
2343 unsigned src_index = use->src_index; in rogue_src_reg_replace() local
2442 unsigned src_index = use->src_index; in rogue_src_regarray_replace() local
[all …]
H A Drogue_compile.c539 unsigned src_index = intr->src[0].ssa->index; in trans_nir_intrinsic_load_global_constant() local
627 unsigned src_index[2] = { alu->src[0].src.ssa->index, in trans_nir_alu_iadd64() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DUpSample.h317 const int64_t src_index = in nearest_neighbor_compute_source_index() local
329 const int64_t src_index = in nearest_neighbor_exact_compute_source_index() local
/aosp_15_r20/external/mesa3d/src/gallium/auxiliary/tgsi/
H A Dtgsi_transform.h496 unsigned src_index, in tgsi_transform_kill_inst()
522 unsigned src_index, in tgsi_transform_tex_inst()
H A Dtgsi_scan.c82 unsigned src_index, in scan_src_operand()
/aosp_15_r20/external/mesa3d/src/asahi/compiler/
H A Dagx_lower_uniform_sources.c16 agx_instr_accepts_uniform(enum agx_opcode op, unsigned src_index, in agx_instr_accepts_uniform()
H A Dagx_nir_opt_preamble.c34 unsigned src_index = ~0; in all_uses_float() local
/aosp_15_r20/external/tensorflow/tensorflow/java/src/main/native/
H A Dutils_jni.cc21 jintArray src_index, TF_Output* dst, jint n) { in resolveOutputs()
/aosp_15_r20/external/mesa3d/src/gallium/drivers/softpipe/
H A Dsp_setup.h48 int src_index:8; member
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dliteral.cc416 int64_t src_index = linear_index(src_literal.shape(), src_indexes); in CopySliceFromInternal() local
434 const LiteralSlice& src_literal, absl::Span<const int64_t> src_index, in CopyElementFrom()
535 ShapeIndex src_index = {i}; in DecomposeTuple() local
754 [&](const ShapeIndex& src_index, Piece* src_piece) { in MoveFrom()
/aosp_15_r20/trusty/user/app/storage/
Dblock_tree.c214 unsigned int src_index, in block_tree_node_shift()
397 unsigned int src_index, in block_tree_node_shift_down()
2405 unsigned int src_index; in block_tree_node_merge() local
/aosp_15_r20/external/mesa3d/src/etnaviv/isa/
H A Dparser.rs188 let mut src_index = 0; in process() localVariable
/aosp_15_r20/external/mesa3d/src/compiler/nir/
H A Dnir_legacy.c30 unsigned src_index = alu_src - alu->src; in nir_legacy_float_mod_folds() local
H A Dnir_opt_intrinsics.c280 unsigned src_index = alu_src - alu->src; in try_opt_exclusive_scan_to_inclusive() local
/aosp_15_r20/external/tensorflow/tensorflow/core/graph/
H A Dnode_builder.cc56 NodeBuilder& NodeBuilder::Input(Node* src_node, int src_index) { in Input()
/aosp_15_r20/external/mesa3d/src/gallium/drivers/llvmpipe/
H A Dlp_bld_interp.h72 uint src_index:8; /* where to find values in incoming vertices */ member
/aosp_15_r20/external/mesa3d/src/gallium/auxiliary/gallivm/
H A Dlp_bld_tgsi.c291 int src_index = get_src_chan_idx(inst->Instruction.Opcode, chan_index); in lp_build_tgsi_inst_llvm() local

123