Searched refs:AddDstTensor (Results 1 – 25 of 36) sorted by relevance
12
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | conv_weights_converter.cc | 128 AddDstTensor("dst_tensor0", op_def.dst_tensors[0]); in GetConverterToConvWeightsCode()
129 AddDstTensor("dst_tensor1", op_def.dst_tensors[1]); in GetConverterToConvWeightsCode()
130 AddDstTensor("dst_tensor2", op_def.dst_tensors[2]); in GetConverterToConvWeightsCode()
131 AddDstTensor("dst_tensor3", op_def.dst_tensors[3]); in GetConverterToConvWeightsCode()
141 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 | split.cc | 34 AddDstTensor("dst_tensor_" + std::to_string(i), definition_.dst_tensors[i]); in GetSplitCode()
86 AddDstTensor("dst_tensor_" + std::to_string(i), definition_.dst_tensors[i]); in GetSplitChannelsCode()
|
| 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 | 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 | space_to_depth.cc | 77 op.AddDstTensor("dst_tensor", op_def.dst_tensors[0]); in CreateSpaceToDepth()
|
| D | reshape.cc | 90 op.AddDstTensor("dst_tensor", definition.dst_tensors[0]); in CreateReshape()
|
| D | winograd.cc | 217 desc.AddDstTensor("dst_tensor", definition.dst_tensors[0]); in CreateWinograd4x4To36()
281 AddDstTensor("dst_tensor", op_def.dst_tensors[0]); in GetWinograd4x4To36TileX6Code()
505 desc.AddDstTensor("dst_tensor", definition.dst_tensors[0]); in CreateWinograd36To4x4()
551 AddDstTensor("dst_tensor", op_def.dst_tensors[0]); in GetWinograd36To4x4Tile4x1Code()
|
| D | resize.cc | 51 AddDstTensor("dst_tensor", dst_desc); in GetResizeCode()
186 AddDstTensor("dst_tensor", dst_desc); in GetResize3DCode()
|
| D | transpose.cc | 103 op.AddDstTensor("dst_tensor", definition.dst_tensors[0]); in CreateTranspose()
|
| D | softmax1x1.cc | 43 AddDstTensor("dst_tensor", op_def.dst_tensors[0]); in GetSoftmaxKernelCode()
|
| D | mean_stddev_normalization.cc | 154 AddDstTensor("dst_tensor", definition_.dst_tensors[0]); in GetNormalizationCode()
|
| D | padding.cc | 30 op->AddDstTensor("dst_tensor", op_def.dst_tensors[0]); in GetPaddingCode()
|
| D | fully_connected.cc | 84 AddDstTensor("dst_tensor", op_def.dst_tensors[0]); in GetFullyConnectedKernelCode()
|
| 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 | convolution_transposed_3x3_thin.cc | 46 AddDstTensor("dst_tensor", op_def.dst_tensors[0]); in GenerateConvolutionTransposedCode()
|
| D | depthwise_conv_3x3.cc | 60 AddDstTensor("dst_tensor", op_def.dst_tensors[0]); in GenerateDepthwiseConvCode()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/ |
| D | fc_fc_add.cc | 77 AddDstTensor("dst_tensor", op_def.dst_tensors[0]); in GetFCFCAddKernelCode()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| D | gpu_operation.h | 132 void AddDstTensor(const std::string& tensor_name,
|
12