| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | split.cc | 32 AddSrcTensor("src_tensor", definition_.src_tensors[0]); in GetSplitCode()
33 for (int i = 0; i < definition_.dst_tensors.size(); ++i) { in GetSplitCode()
34 AddDstTensor("dst_tensor_" + std::to_string(i), definition_.dst_tensors[i]); in GetSplitCode()
51 if (definition_.src_tensors[0].HasAxis(Axis::BATCH)) { in GetSplitCode()
58 if (definition_.src_tensors[0].HasAxis(Axis::DEPTH)) { in GetSplitCode()
71 for (int i = 0; i < definition_.dst_tensors.size(); ++i) { in GetSplitCode()
84 AddSrcTensor("src_tensor", definition_.src_tensors[0]); in GetSplitChannelsCode()
85 for (int i = 0; i < definition_.dst_tensors.size(); ++i) { in GetSplitChannelsCode()
86 AddDstTensor("dst_tensor_" + std::to_string(i), definition_.dst_tensors[i]); in GetSplitChannelsCode()
90 definition_.src_tensors[0].HasAxis(Axis::BATCH) ? ", B" : ""; in GetSplitChannelsCode()
[all …]
|
| D | mean_stddev_normalization.cc | 153 AddSrcTensor("src_tensor", definition_.src_tensors[0]); in GetNormalizationCode()
154 AddDstTensor("dst_tensor", definition_.dst_tensors[0]); in GetNormalizationCode()
|
| D | winograd.cc | 233 code_ = GetWinograd4x4To36TileX6Code(definition_); in Winograd4x4To36TileX6()
237 if (definition_.precision == CalculationsPrecision::F16 && in Winograd4x4To36TileX6()
433 desc.element_type = definition_.GetDataType(); in UploadBt()
522 if (definition_.precision == CalculationsPrecision::F16 && in Winograd36To4x4Tile4x1()
526 code_ = GetWinograd36To4x4Tile4x1Code(definition_); in Winograd36To4x4Tile4x1()
652 desc.element_type = definition_.GetDataType(); in UploadAt()
|
| D | convolution_transposed_thin.cc | 32 definition_, DivideRoundUp(attr.weights.shape.i, 4), attr.weights.shape.o, in ConvolutionTransposedThin()
34 if (definition_.precision == CalculationsPrecision::F16 && in ConvolutionTransposedThin()
|
| D | convolution_transposed_4x4.cc | 63 code_ = GenerateConvolutionTransposedCode(gpu_info, definition_, in ConvolutionTransposed4x4()
65 if (definition_.precision == CalculationsPrecision::F16 && in ConvolutionTransposed4x4()
397 DataType weights_type = definition_.precision == CalculationsPrecision::F32 in UploadWeights()
|
| D | resize.cc | 26 code_ = GetResizeCode(definition_, attr_); in Resize()
161 code_ = GetResize3DCode(definition_, attr_); in Resize3D()
|
| D | conv_buffer_1x1.h | 130 const bool f32_weights = definition_.precision == CalculationsPrecision::F32; in UploadWeights()
162 desc.element_type = definition_.GetDataType(); in UploadBiases()
|
| D | convolution_transposed_3x3.cc | 46 code_ = GenerateConvolutionTransposedCode(gpu_info, definition_, in ConvolutionTransposed3x3()
49 if (definition_.precision == CalculationsPrecision::F16 && in ConvolutionTransposed3x3()
392 DataType weights_type = definition_.precision == CalculationsPrecision::F32 in UploadWeights()
|
| D | convolution_transposed.h | 102 DataType weights_type = definition_.precision == CalculationsPrecision::F32 in UploadWeights()
152 const bool f32_weights = definition_.precision == CalculationsPrecision::F32; in UploadWeights()
|
| D | convolution_transposed_3x3_thin.cc | 37 definition_, DivideRoundUp(attr.weights.shape.i, 4), in ConvolutionTransposed3x3Thin()
208 DataType weights_type = definition_.precision == CalculationsPrecision::F32 in UploadWeights()
|
| D | depthwise_conv_3x3.cc | 33 code_ = GenerateDepthwiseConvCode(definition_, weights_are_buffer, in DepthwiseConv3x3()
36 if (definition_.precision == CalculationsPrecision::F16 && in DepthwiseConv3x3()
|
| D | softmax1x1.cc | 29 code_ = GetSoftmaxKernelCode(definition_); in Softmax1x1()
|
| D | fully_connected.cc | 57 code_ = GetFullyConnectedKernelCode(definition_, gpu_info); in FullyConnected()
|
| D | convolution_transposed.cc | 73 code_ = GenerateConvolutionTransposedCode(definition_, gpu_info, in ConvolutionTransposed()
123 code_ = GenerateConvolutionTransposedCode(definition_, gpu_info, in ConvolutionTransposed()
551 if (definition_.src_tensors[0].HasAxis(Axis::DEPTH)) { in BindArguments()
|
| D | strided_slice.cc | 81 code_ = GetStridedSliceCode(definition_, Is4Aligned(attributes_)); in StridedSlice()
|
| D | depthwise_conv_3x3_stride_h2.h | 78 const bool fp32_weights = definition_.precision == CalculationsPrecision::F32; in UploadWeightsAndBiases()
|
| D | convolution_transposed_thin.h | 74 const bool f32_weights = definition_.precision == CalculationsPrecision::F32; in UploadData()
|
| D | depthwise_conv_3x3.h | 84 const bool fp32_weights = definition_.precision == CalculationsPrecision::F32; in UploadWeightsAndBiases()
|
| D | fully_connected.h | 154 const bool f32_weights = definition_.precision == CalculationsPrecision::F32; in UploadWeights()
|
| D | conv_weights_converter.cc | 30 code_ = GetConverterToConvWeightsCode(definition_, weights_desc_); in ConverterToConvWeights()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| D | gpu_operation.cc | 76 : definition_(definition) {} in GPUOperation()
101 definition_(std::move(operation.definition_)), in GPUOperation()
123 definition_ = std::move(operation.definition_); in operator =()
146 definition_.src_tensors.push_back( in AddOperation()
147 operation->definition_.src_tensors[i + 1]); in AddOperation()
189 absl::make_unique<TensorDescriptor>(definition_.src_tensors[0]); in AssembleCode()
190 if (definition_.IsBatchSupported()) { in AssembleCode()
197 absl::make_unique<TensorDescriptor>(definition_.dst_tensors[0]); in AssembleCode()
198 if (definition_.IsBatchSupported()) { in AssembleCode()
205 code_ = GetElementWiseCode(definition_, check_src_channels_size_); in AssembleCode()
|
| D | gpu_operation.h | 124 const OperationDef& GetDefinition() const { return definition_; } in GetDefinition()
168 OperationDef definition_; variable
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/ |
| D | compute_task.cc | 70 return operation_->definition_; in GetDefinition()
88 return CompileProgram(device, operation_->definition_.precision, in Compile()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | cl_operation.h | 61 const OperationDef& GetDefinition() const { return operation_->definition_; } in GetDefinition()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/ |
| D | fc_fc_add.cc | 55 code_ = GetFCFCAddKernelCode(definition_, gpu_info); in FCFCAdd()
|