Home
last modified time | relevance | path

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

12

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
H A Dconv_weights_converter.cc152 AddDstTensor("dst_tensor0", definition_.dst_tensors[0]); in GetConverterToConvWeightsCode()
153 AddDstTensor("dst_tensor1", definition_.dst_tensors[1]); in GetConverterToConvWeightsCode()
154 AddDstTensor("dst_tensor2", definition_.dst_tensors[2]); in GetConverterToConvWeightsCode()
155 AddDstTensor("dst_tensor3", definition_.dst_tensors[3]); in GetConverterToConvWeightsCode()
164 AddDstTensor("dst_tensor", definition_.dst_tensors[0]); in GetConverterToConvWeightsCode()
H A Dlstm.cc93 op.AddDstTensor("new_state", definition.dst_tensors[0]); in CreateLSTM()
94 op.AddDstTensor("activation", definition.dst_tensors[1]); in CreateLSTM()
H A Dpooling.cc35 op->AddDstTensor("dst_tensor", op_def.dst_tensors[0]); in GetAveragePoolingKernelCode()
132 op->AddDstTensor("dst_tensor", op_def.dst_tensors[0]); in GetMaxPoolingKernelCode()
134 op->AddDstTensor("dst_indices", op_def.dst_tensors[1]); in GetMaxPoolingKernelCode()
H A Dspace_to_depth.cc117 op.AddDstTensor("dst_tensor", op_def.dst_tensors[0]); in CreateSpaceToDepth()
128 op.AddDstTensor("dst_tensor", op_def.dst_tensors[0]); in CreateDepthToSpace()
H A Dwinograd.cc256 desc.AddDstTensor("dst_tensor", definition.dst_tensors[0]); in CreateWinograd4x4To36()
293 AddDstTensor("dst_tensor", op_def.dst_tensors[0]); in GetWinograd4x4To36TileX6Code()
551 desc.AddDstTensor("dst_tensor", definition.dst_tensors[0]); in CreateWinograd36To4x4()
585 AddDstTensor("dst_tensor", op_def.dst_tensors[0]); in GetWinograd36To4x4Tile4x1Code()
H A Delementwise.cc408 op.AddDstTensor("dst_tensor", definition.dst_tensors[0]); in CreateElementwiseOneInputWithBroadcast()
433 op.AddDstTensor("dst_tensor", definition.dst_tensors[0]); in CreateElementwiseWithBroadcast()
457 op.AddDstTensor("dst_tensor", definition.dst_tensors[0]); in CreateElementwiseTwoInputWithBroadcast()
H A Dresize.cc46 AddDstTensor("dst_tensor", op_def.dst_tensors[0]); in GetResizeCode()
133 AddDstTensor("dst_tensor", op_def.dst_tensors[0]); in GetResize3DCode()
H A Dsplit.cc35 AddDstTensor("dst_tensor_" + std::to_string(i), definition_.dst_tensors[i]); in GetSplitCode()
124 AddDstTensor("dst_tensor_" + std::to_string(i), definition_.dst_tensors[i]); in GetSplitChannelsCode()
H A Dgather.cc58 op.AddDstTensor("dst_tensor", op_def.dst_tensors[0]); in CreateGather()
H A Dreshapex4.cc72 op.AddDstTensor("dst_tensor", definition.dst_tensors[0]); in CreateReshapex4()
H A Dsoftmax.cc77 op.AddDstTensor("dst_tensor", definition.dst_tensors[0]); in CreateSoftmax()
H A Dresampler.cc84 op.AddDstTensor("dst_tensor", definition.dst_tensors[0]); in CreateResampler()
H A Dreshape.cc86 op.AddDstTensor("dst_tensor", definition.dst_tensors[0]); in CreateReshape()
H A Done_hot.cc28 op->AddDstTensor("dst_tensor", op_def.dst_tensors[0]); in GetOneHotCode()
H A Dselect_v2.cc28 op->AddDstTensor("dst_tensor", op_def.dst_tensors[0]); in GetSelectV2Code()
H A Dtile.cc102 op.AddDstTensor("dst_tensor", op_def.dst_tensors[0]); in CreateTile()
H A Dtranspose.cc99 op.AddDstTensor("dst_tensor", definition.dst_tensors[0]); in CreateTranspose()
H A Dadd.cc30 op.AddDstTensor("dst_tensor", op_def.dst_tensors[0]); in CreateUnequalAdd()
H A Dsoftmax1x1.cc44 AddDstTensor("dst_tensor", op_def.dst_tensors[0]); in GetSoftmaxKernelCode()
H A Dpadding.cc30 op->AddDstTensor("dst_tensor", op_def.dst_tensors[0]); in GetPaddingCode()
H A Dcumsum.cc28 AddDstTensor("dst_tensor", op_def.dst_tensors[0]); in GetCumsumCode()
H A Dconcat_xy.cc121 op.AddDstTensor("dst_tensor", definition.dst_tensors[0]); in CreateConcatXY()
H A Dconcat_z.cc144 op.AddDstTensor("dst_tensor", definition.dst_tensors[0]); in CreateConcatZ()
H A Dmax_unpooling.cc36 op->AddDstTensor("dst_tensor", op_def.dst_tensors[0]); in GetMaxUnpoolingKernelCode()
H A Dstrided_slice.cc99 AddDstTensor("dst_tensor", op_def.dst_tensors[0]); in GetStridedSliceCode()

12