Searched refs:AddSrcTensor (Results 1 – 25 of 38) sorted by relevance
12
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | lstm.cc | 91 op.AddSrcTensor("intermediate", definition.src_tensors[0]); in CreateLSTM()
92 op.AddSrcTensor("prev_state", definition.src_tensors[1]); in CreateLSTM()
|
| D | max_unpooling.cc | 33 op->AddSrcTensor("src_tensor", src_desc); in GetMaxUnpoolingKernelCode()
39 op->AddSrcTensor("src_indices", src_ind_desc); in GetMaxUnpoolingKernelCode()
|
| D | split.cc | 32 AddSrcTensor("src_tensor", definition_.src_tensors[0]); in GetSplitCode()
84 AddSrcTensor("src_tensor", definition_.src_tensors[0]); in GetSplitChannelsCode()
|
| D | add.cc | 42 add.AddSrcTensor(tensor_name, src_desc); in CreateAdd()
|
| D | reshapex4.cc | 70 op.AddSrcTensor("src_tensor", definition.src_tensors[0]); in CreateReshapex4()
|
| D | softmax.cc | 72 op.AddSrcTensor("src_tensor", src_desc); in CreateSoftmax()
|
| D | space_to_depth.cc | 76 op.AddSrcTensor("src_tensor", op_def.src_tensors[0]); in CreateSpaceToDepth()
|
| D | reshape.cc | 89 op.AddSrcTensor("src_tensor", definition.src_tensors[0]); in CreateReshape()
|
| D | winograd.cc | 216 desc.AddSrcTensor("src_tensor", definition.src_tensors[0]); in CreateWinograd4x4To36()
280 AddSrcTensor("src_tensor", src_desc); in GetWinograd4x4To36TileX6Code()
504 desc.AddSrcTensor("src_tensor", definition.src_tensors[0]); in CreateWinograd36To4x4()
550 AddSrcTensor("src_tensor", src_desc); in GetWinograd36To4x4Tile4x1Code()
|
| D | resize.cc | 46 AddSrcTensor("src_tensor", src_desc); in GetResizeCode()
181 AddSrcTensor("src_tensor", src_desc); in GetResize3DCode()
|
| D | transpose.cc | 102 op.AddSrcTensor("src_tensor", definition.src_tensors[0]); in CreateTranspose()
|
| D | softmax1x1.cc | 42 AddSrcTensor("src_tensor", op_def.src_tensors[0]); in GetSoftmaxKernelCode()
|
| D | mean_stddev_normalization.cc | 153 AddSrcTensor("src_tensor", definition_.src_tensors[0]); in GetNormalizationCode()
|
| D | padding.cc | 29 op->AddSrcTensor("src_tensor", op_def.src_tensors[0]); in GetPaddingCode()
|
| D | pooling.cc | 35 op->AddSrcTensor("src_tensor", src_desc); in GetAveragePoolingKernelCode()
152 op->AddSrcTensor("src_tensor", src_desc); in GetMaxPoolingKernelCode()
|
| D | fully_connected.cc | 83 AddSrcTensor("src_tensor", op_def.src_tensors[0]); in GetFullyConnectedKernelCode()
|
| D | depthwise_conv.cc | 77 op->AddSrcTensor("src_tensor", src_desc); in GenerateDepthwiseConvolutionCode()
79 op->AddSrcTensor("weights", op_def.src_tensors[1]); in GenerateDepthwiseConvolutionCode()
|
| D | concat_xy.cc | 119 op.AddSrcTensor(name, definition.src_tensors[i]); in CreateConcatXY()
|
| D | concat_z.cc | 132 op.AddSrcTensor(name, src_desc); in CreateConcatZ()
|
| D | strided_slice.cc | 97 AddSrcTensor("src_tensor", op_def.src_tensors[0]); in GetStridedSliceCode()
|
| D | conv_weights_converter.cc | 49 AddSrcTensor("src_tensor", op_def.src_tensors[0]); in GetConverterToConvWeightsCode()
|
| D | depthwise_conv_3x3_stride_h2.cc | 222 desc.AddSrcTensor("src_tensor", src_desc); in CreateDepthWiseConv3x3StrideH2()
|
| D | convolution_transposed_thin.cc | 55 AddSrcTensor("src_tensor", op_def.src_tensors[0]); in GenerateConvolutionTransposedCode()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/ |
| D | fc_fc_add.cc | 75 AddSrcTensor("src_tensor_0", op_def.src_tensors[0]); in GetFCFCAddKernelCode()
76 AddSrcTensor("src_tensor_1", op_def.src_tensors[1]); in GetFCFCAddKernelCode()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| D | gpu_operation.h | 126 void AddSrcTensor(const std::string& tensor_name,
|
12