Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_relu_op.cc441 const size_t src_index = 0; // index of src input tensor in Compute() local
591 size_t src_index = GetSrcIndex(); in Compute() local
771 const size_t src_index = 0; // index of src input tensor in Compute_Scalar() local
803 const size_t src_index = 1; // index of src input tensor in Compute_Scalar() local
839 const size_t src_index = 0; // index of src input tensor in Compute_Scalar() local
875 const size_t src_index = 1; // index of src input tensor in Compute_Scalar() local
920 const size_t src_index = 0; // index of src input tensor in Compute_Scalar() local
966 const size_t src_index = GetSrcIndex(); in Compute_Scalar() local
1005 const size_t src_index = 0; // index of src input tensor in Compute_Scalar() local
1037 const size_t src_index = 1; // index of src input tensor in Compute_Scalar() local
[all …]
Dmkl_conv_ops.h286 virtual inline void GetFilterSizeInMklOrder(size_t src_index, in GetFilterSizeInMklOrder()
486 size_t src_index, size_t filter_index, const memory::dims& strides, in GetOutputAndPadSizeInMklOrder()
Dmkl_aggregate_ops.cc42 TensorShape GetTensorShape(OpKernelContext* ctx, size_t src_index) { in GetTensorShape()
/external/mesa3d/src/gallium/auxiliary/draw/
Ddraw_vertex.h78 unsigned src_index:8; /**< map to post-xform attribs */ member
115 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/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.cc238 ShapeIndex src_index; in HandleGetTupleElement() local
352 const PointsToSet::BufferList& points_to) { in HandleCopyDone()
390 const PointsToSet::BufferList& points_to) { in HandleSend()
424 const PointsToSet::BufferList& points_to) { in HandleTuple()
/external/mesa3d/src/gallium/drivers/r300/compiler/
Dradeon_optimize.c486 unsigned int src_index; in presub_helper() local
504 unsigned int src_index) in presub_replace_add()
619 unsigned int src_index) in presub_replace_inv()
Dradeon_variable.c87 int src_index = reader->U.P.Arg->Source; in rc_variable_change_dst() local
/external/tensorflow/tensorflow/core/graph/
Dnode_builder.cc51 NodeBuilder& NodeBuilder::Input(Node* src_node, int src_index) { in Input()
/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/gallium/auxiliary/tgsi/
Dtgsi_transform.h524 unsigned src_index, in tgsi_transform_kill_inst()
550 unsigned src_index, in tgsi_transform_tex_inst()
Dtgsi_scan.c110 unsigned src_index, in scan_src_operand()
/external/mesa3d/src/gallium/auxiliary/gallivm/
Dlp_bld_tgsi.c286 int src_index = get_src_chan_idx(inst->Instruction.Opcode, chan_index); in lp_build_tgsi_inst_llvm() local
/external/tensorflow/tensorflow/compiler/xla/
Dliteral.cc311 int64 src_index = linear_index(src_literal.shape(), src_indexes); in CopySliceFromInternal() local
329 absl::Span<const int64> src_index, in CopyElementFrom()
414 ShapeIndex src_index = {i}; in DecomposeTuple() local
610 [&](const ShapeIndex& src_index, const Piece& src_piece) { in MoveFrom()
/external/tensorflow/tensorflow/core/lib/gtl/
Dflatrep.h320 void FreshInsert(Bucket* src, uint32 src_index, Copier copier) { in FreshInsert()
/external/mesa3d/src/gallium/drivers/etnaviv/
Detnaviv_compiler_nir.h84 src_index(nir_function_impl *impl, nir_src *src) in src_index() function
/external/webrtc/modules/audio_processing/utility/
Ddelay_estimator.cc346 int src_index = 0; in WebRtc_SoftResetBinaryDelayEstimatorFarend() local
/external/mesa3d/src/gallium/drivers/r300/compiler/tests/
Drc_test_helpers.c136 unsigned int src_index, in init_rc_normal_src()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_compile_op_support.cc156 xla::ShapeIndex src_index; in GetSubtree() local
/external/mesa3d/src/compiler/nir/
Dnir_opt_vectorize.c283 unsigned src_index = 5; in instr_try_combine() local
/external/mesa3d/src/panfrost/midgard/
Dmidgard_ra.c727 unsigned src_index = ins->src[i]; in install_registers_instr() local
/external/tensorflow/tensorflow/compiler/tf2xla/
Drearrange_function_argument.cc202 int src_index; in CalculateRetvalRearrange() local
/external/tensorflow/tensorflow/core/common_runtime/
Dgraph_constructor.cc387 int src_index; member
1178 int src_index; in Convert() local

12