Home
last modified time | relevance | path

Searched refs:AddSrcTensor (Results 1 – 25 of 44) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
H A Dselect_v2.cc25 op->AddSrcTensor("cond_tensor", op_def.src_tensors[0]); in GetSelectV2Code()
26 op->AddSrcTensor("true_tensor", op_def.src_tensors[1]); in GetSelectV2Code()
27 op->AddSrcTensor("else_tensor", op_def.src_tensors[2]); in GetSelectV2Code()
H A Delementwise.cc407 op.AddSrcTensor("src_tensor", definition.src_tensors[0]); in CreateElementwiseOneInputWithBroadcast()
432 op.AddSrcTensor("src_tensor", definition.src_tensors[0]); in CreateElementwiseWithBroadcast()
455 op.AddSrcTensor("src0_tensor", definition.src_tensors[0]); in CreateElementwiseTwoInputWithBroadcast()
456 op.AddSrcTensor("src1_tensor", definition.src_tensors[1]); in CreateElementwiseTwoInputWithBroadcast()
H A Dgather.cc56 op.AddSrcTensor("src_tensor", op_def.src_tensors[0]); in CreateGather()
57 op.AddSrcTensor("indices", op_def.src_tensors[1]); in CreateGather()
H A Dresampler.cc82 op.AddSrcTensor("src_tensor", definition.src_tensors[0]); in CreateResampler()
83 op.AddSrcTensor("warp_tensor", definition.src_tensors[1]); in CreateResampler()
H A Dlstm.cc91 op.AddSrcTensor("intermediate", definition.src_tensors[0]); in CreateLSTM()
92 op.AddSrcTensor("prev_state", definition.src_tensors[1]); in CreateLSTM()
H A Dspace_to_depth.cc116 op.AddSrcTensor("src_tensor", op_def.src_tensors[0]); in CreateSpaceToDepth()
127 op.AddSrcTensor("src_tensor", op_def.src_tensors[0]); in CreateDepthToSpace()
H A Dwinograd.cc255 desc.AddSrcTensor("src_tensor", definition.src_tensors[0]); in CreateWinograd4x4To36()
292 AddSrcTensor("src_tensor", op_def.src_tensors[0]); in GetWinograd4x4To36TileX6Code()
550 desc.AddSrcTensor("src_tensor", definition.src_tensors[0]); in CreateWinograd36To4x4()
584 AddSrcTensor("src_tensor", op_def.src_tensors[0]); in GetWinograd36To4x4Tile4x1Code()
H A Dmax_unpooling.cc34 op->AddSrcTensor("src_tensor", op_def.src_tensors[0]); in GetMaxUnpoolingKernelCode()
35 op->AddSrcTensor("src_indices", op_def.src_tensors[1]); in GetMaxUnpoolingKernelCode()
H A Dresize.cc45 AddSrcTensor("src_tensor", op_def.src_tensors[0]); in GetResizeCode()
132 AddSrcTensor("src_tensor", op_def.src_tensors[0]); in GetResize3DCode()
H A Dsplit.cc33 AddSrcTensor("src_tensor", definition_.src_tensors[0]); in GetSplitCode()
122 AddSrcTensor("src_tensor", definition_.src_tensors[0]); in GetSplitChannelsCode()
H A Dpooling.cc34 op->AddSrcTensor("src_tensor", op_def.src_tensors[0]); in GetAveragePoolingKernelCode()
131 op->AddSrcTensor("src_tensor", op_def.src_tensors[0]); in GetMaxPoolingKernelCode()
H A Dreshapex4.cc71 op.AddSrcTensor("src_tensor", definition.src_tensors[0]); in CreateReshapex4()
H A Dsoftmax.cc76 op.AddSrcTensor("src_tensor", definition.src_tensors[0]); in CreateSoftmax()
H A Dreshape.cc85 op.AddSrcTensor("src_tensor", definition.src_tensors[0]); in CreateReshape()
H A Done_hot.cc27 op->AddSrcTensor("src_tensor", op_def.src_tensors[0]); in GetOneHotCode()
H A Dtile.cc101 op.AddSrcTensor("src_tensor", op_def.src_tensors[0]); in CreateTile()
H A Dtranspose.cc98 op.AddSrcTensor("src_tensor", definition.src_tensors[0]); in CreateTranspose()
H A Dadd.cc33 op.AddSrcTensor(tensor_name, op_def.src_tensors[i]); in CreateUnequalAdd()
H A Dsoftmax1x1.cc43 AddSrcTensor("src_tensor", op_def.src_tensors[0]); in GetSoftmaxKernelCode()
H A Ddepthwise_conv.cc221 AddSrcTensor("src_tensor", definition_.src_tensors[0]); in GenerateCode()
223 AddSrcTensor("weights", definition_.src_tensors[1]); in GenerateCode()
H A Dpadding.cc29 op->AddSrcTensor("src_tensor", op_def.src_tensors[0]); in GetPaddingCode()
H A Dcumsum.cc27 AddSrcTensor("src_tensor", op_def.src_tensors[0]); in GetCumsumCode()
H A Dconcat_xy.cc119 op.AddSrcTensor(name, definition.src_tensors[i]); in CreateConcatXY()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/
H A Dconv_pointwise.cc195 op.AddSrcTensor("src_tensor", definition.src_tensors[0]); in CreateConvPointwise()
196 op.AddSrcTensor("weights_tensor", definition.src_tensors[1]); in CreateConvPointwise()
H A Dfc_fc_add.cc106 AddSrcTensor("src_tensor_0", op_def.src_tensors[0]); in GetFCFCAddKernelCode()
107 AddSrcTensor("src_tensor_1", op_def.src_tensors[1]); in GetFCFCAddKernelCode()

12