Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/core/kernels/mkl/
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 …]
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()
Dmkl_aggregate_ops.cc41 TensorShape GetTensorShape(OpKernelContext* ctx, size_t src_index) { in GetTensorShape()
/external/mesa3d/src/gallium/auxiliary/draw/
Ddraw_vertex.h77 unsigned src_index:8; /**< map to post-xform attribs */ member
117 int src_index) in draw_emit_vertex_attr()
/external/tensorflow/tensorflow/core/framework/
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()
/external/mesa3d/src/asahi/compiler/
Dagx_lower_uniform_sources.c16 should_lower(enum agx_opcode op, agx_index uniform, unsigned src_index) in should_lower()
/external/tensorflow/tensorflow/java/src/main/native/
Dutils_jni.cc21 jintArray src_index, TF_Output* dst, jint n) { in resolveOutputs()
/external/mesa3d/src/gallium/drivers/softpipe/
Dsp_setup.h48 int src_index:8; member
/external/tensorflow/tensorflow/compiler/xla/service/
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()
/external/mesa3d/src/virtio/vulkan/
Dvn_query_pool.c194 uint32_t src_index = 0; in vn_get_query_pool_feedback() local
269 uint32_t src_index = 0; in vn_query_feedback_wait_ready() local
/external/mesa3d/src/gallium/drivers/r300/compiler/
Dradeon_optimize.c400 unsigned int src_index; in presub_helper() local
416 unsigned int src_index) in presub_replace_add()
537 unsigned int src_index) in presub_replace_inv()
554 unsigned int src_index) in presub_replace_bias()
Dradeon_variable.c88 int src_index = reader->U.P.Arg->Source; in rc_variable_change_dst() local
/external/tensorflow/tensorflow/core/graph/
Dnode_builder.cc56 NodeBuilder& NodeBuilder::Input(Node* src_node, int src_index) { in Input()
/external/mesa3d/src/compiler/nir/
Dnir_legacy.c30 unsigned src_index = alu_src - alu->src; in nir_legacy_float_mod_folds() local
Dnir_opt_intrinsics.c269 unsigned src_index = alu_src - alu->src; in try_opt_exclusive_scan_to_inclusive() local
/external/mesa3d/src/gallium/auxiliary/tgsi/
Dtgsi_scan.c82 unsigned src_index, in scan_src_operand()
Dtgsi_transform.h496 unsigned src_index, in tgsi_transform_kill_inst()
522 unsigned src_index, in tgsi_transform_tex_inst()
/external/mesa3d/src/freedreno/ir3/
Dir3_nir_opt_preamble.c58 unsigned src_index = ~0; in all_uses_float() local
/external/mesa3d/src/gallium/drivers/llvmpipe/
Dlp_bld_interp.h72 uint src_index:8; /* where to find values in incoming vertices */ member
/external/mesa3d/src/imagination/rogue/
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
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 …]
/external/mesa3d/src/gallium/auxiliary/gallivm/
Dlp_bld_tgsi.c291 int src_index = get_src_chan_idx(inst->Instruction.Opcode, chan_index); in lp_build_tgsi_inst_llvm() local
/external/tensorflow/tensorflow/core/lib/gtl/
Dflatrep.h320 void FreshInsert(Bucket* src, uint32 src_index, Copier copier) { in FreshInsert()
/external/webrtc/modules/audio_processing/utility/
Ddelay_estimator.cc346 int src_index = 0; in WebRtc_SoftResetBinaryDelayEstimatorFarend() local
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_compile_op_support.cc158 xla::ShapeIndex src_index; in GetSubtree() local

123