Searched refs:output_data_tensor (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| H A D | conv_2d_builder.cc | 127 const TfLiteTensor& output_data_tensor, OpBuilder* data_min_const, in BuildDilatedDwConv() argument
138 &output_depth_size, output_data_tensor.dims); in BuildDilatedDwConv()
241 const TfLiteTensor& output_data_tensor, OpBuilder* data_min_const, in BuildSplittedDwConv() argument
254 &output_depth_size, output_data_tensor.dims); in BuildSplittedDwConv()
329 const TfLiteIntArray* inputs, const TfLiteTensor& output_data_tensor, in BuildStandardConv() argument
340 &output_depth_size, output_data_tensor.dims); in BuildStandardConv()
376 const auto& output_data_tensor = context->tensors[outputs->data[0]]; in PopulateSubGraph() local
479 BuildDilatedDwConv(inputs, data_tensor, output_data_tensor, data_min_const, in PopulateSubGraph()
485 BuildSplittedDwConv(inputs, data_tensor, output_data_tensor, data_min_const, in PopulateSubGraph()
490 BuildStandardConv(inputs, output_data_tensor, data_min_const, in PopulateSubGraph()
|
| H A D | conv_2d_builder.h | 67 const TfLiteTensor& output_data_tensor,
77 const TfLiteTensor& output_data_tensor,
87 const TfLiteTensor& output_data_tensor, OpBuilder* data_min_const,
|