Searched refs:AddDstTensor (Results 1 – 25 of 39) sorted by relevance
12
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | conv_weights_converter.cc | 136 AddDstTensor("dst_tensor0", op_def.dst_tensors[0]); in GetConverterToConvWeightsCode()
137 AddDstTensor("dst_tensor1", op_def.dst_tensors[1]); in GetConverterToConvWeightsCode()
138 AddDstTensor("dst_tensor2", op_def.dst_tensors[2]); in GetConverterToConvWeightsCode()
139 AddDstTensor("dst_tensor3", op_def.dst_tensors[3]); in GetConverterToConvWeightsCode()
149 AddDstTensor("dst_tensor", op_def.dst_tensors[0]); in GetConverterToConvWeightsCode()
|
| D | lstm.cc | 93 op.AddDstTensor("new_state", definition.dst_tensors[0]); in CreateLSTM()
94 op.AddDstTensor("activation", definition.dst_tensors[1]); in CreateLSTM()
|
| D | space_to_depth.cc | 121 op.AddDstTensor("dst_tensor", op_def.dst_tensors[0]); in CreateSpaceToDepth()
132 op.AddDstTensor("dst_tensor", op_def.dst_tensors[0]); in CreateDepthToSpace()
|
| D | pooling.cc | 40 op->AddDstTensor("dst_tensor", dst_desc); in GetAveragePoolingKernelCode()
157 op->AddDstTensor("dst_tensor", dst_desc); in GetMaxPoolingKernelCode()
163 op->AddDstTensor("dst_indices", dst_ind_desc); in GetMaxPoolingKernelCode()
|
| D | gather.cc | 58 op.AddDstTensor("dst_tensor", op_def.dst_tensors[0]); in CreateGather()
|
| D | reshapex4.cc | 71 op.AddDstTensor("dst_tensor", definition.dst_tensors[0]); in CreateReshapex4()
|
| D | softmax.cc | 77 op.AddDstTensor("dst_tensor", dst_desc); in CreateSoftmax()
|
| D | split.cc | 34 AddDstTensor("dst_tensor_" + std::to_string(i), definition_.dst_tensors[i]); in GetSplitCode()
121 AddDstTensor("dst_tensor_" + std::to_string(i), definition_.dst_tensors[i]); in GetSplitChannelsCode()
|
| D | resampler.cc | 80 op.AddDstTensor("dst_tensor", definition.dst_tensors[0]); in CreateResampler()
|
| D | reshape.cc | 89 op.AddDstTensor("dst_tensor", definition.dst_tensors[0]); in CreateReshape()
|
| D | resize.cc | 54 AddDstTensor("dst_tensor", dst_desc); in GetResizeCode()
189 AddDstTensor("dst_tensor", dst_desc); in GetResize3DCode()
|
| D | winograd.cc | 224 desc.AddDstTensor("dst_tensor", definition.dst_tensors[0]); in CreateWinograd4x4To36()
291 AddDstTensor("dst_tensor", op_def.dst_tensors[0]); in GetWinograd4x4To36TileX6Code()
542 desc.AddDstTensor("dst_tensor", definition.dst_tensors[0]); in CreateWinograd36To4x4()
588 AddDstTensor("dst_tensor", op_def.dst_tensors[0]); in GetWinograd36To4x4Tile4x1Code()
|
| D | tile.cc | 103 op.AddDstTensor("dst_tensor", op_def.dst_tensors[0]); in CreateTile()
|
| D | transpose.cc | 102 op.AddDstTensor("dst_tensor", definition.dst_tensors[0]); in CreateTranspose()
|
| D | softmax1x1.cc | 44 AddDstTensor("dst_tensor", op_def.dst_tensors[0]); in GetSoftmaxKernelCode()
|
| D | padding.cc | 30 op->AddDstTensor("dst_tensor", op_def.dst_tensors[0]); in GetPaddingCode()
|
| D | mean_stddev_normalization.cc | 170 AddDstTensor("dst_tensor", definition_.dst_tensors[0]); in GetNormalizationCode()
|
| D | concat_xy.cc | 121 op.AddDstTensor("dst_tensor", definition.dst_tensors[0]); in CreateConcatXY()
|
| D | concat_z.cc | 138 op.AddDstTensor("dst_tensor", dst_desc); in CreateConcatZ()
|
| D | max_unpooling.cc | 44 op->AddDstTensor("dst_tensor", dst_desc); in GetMaxUnpoolingKernelCode()
|
| D | strided_slice.cc | 98 AddDstTensor("dst_tensor", op_def.dst_tensors[0]); in GetStridedSliceCode()
|
| D | convolution_transposed_thin.cc | 56 AddDstTensor("dst_tensor", op_def.dst_tensors[0]); in GenerateConvolutionTransposedCode()
|
| D | depthwise_conv_3x3_stride_h2.cc | 223 desc.AddDstTensor("dst_tensor", definition.dst_tensors[0]); in CreateDepthWiseConv3x3StrideH2()
|
| D | fully_connected.cc | 114 AddDstTensor("dst_tensor", op_def.dst_tensors[0]); in GetFullyConnectedKernelCode()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| D | gpu_operation.h | 138 void AddDstTensor(const std::string& tensor_name,
|
12