Searched refs:output_ids (Results 1 – 10 of 10) sorted by relevance
108 const flatbuffers::Vector<int32_t> *output_ids() const { in output_ids() function121 verifier.VerifyVector(output_ids()) && in Verify()138 void add_output_ids(flatbuffers::Offset<flatbuffers::Vector<int32_t>> output_ids) { in add_output_ids()139 fbb_.AddOffset(CLNode::VT_OUTPUT_IDS, output_ids); in add_output_ids()159 flatbuffers::Offset<flatbuffers::Vector<int32_t>> output_ids = 0,163 builder_.add_output_ids(output_ids);173 const std::vector<int32_t> *output_ids = nullptr,176 auto output_ids__ = output_ids ? _fbb.CreateVector<int32_t>(*output_ids) : 0;293 const flatbuffers::Vector<int32_t> *output_ids() const { in output_ids() function324 verifier.VerifyVector(output_ids()) && in Verify()[all …]
27 output_ids:[int32];48 output_ids:[int32];
386 cl_node.outputs.resize(gpu_op.output_ids.size()); in ConvertOperations()387 for (int j = 0; j < gpu_op.output_ids.size(); ++j) { in ConvertOperations()388 int id = gpu_op.output_ids[j]; in ConvertOperations()
945 for (auto out_fb : *fb_node->output_ids()) { in Decode()1052 for (auto out_fb : *fb_inference->output_ids()) { in Decode()
121 winograd_up.output_ids = {-1}; in WinogradFromNode()129 conv.output_ids = {-2}; in WinogradFromNode()139 winograd_down.output_ids = {static_cast<int>(outputs[0]->id)}; in WinogradFromNode()224 conv_op.output_ids = {static_cast<int>(outputs[0]->id)}; in GPUOperationFromNode()246 converter_op.output_ids = {-1}; in GPUOperationFromNode()256 transpose_op.output_ids = {-2}; in GPUOperationFromNode()301 conv_op.output_ids = {static_cast<int>(outputs[0]->id)}; in GPUOperationFromNode()323 converter_op.output_ids = {-1}; in GPUOperationFromNode()351 conv_op.output_ids = {static_cast<int>(outputs[0]->id)}; in GPUOperationFromNode()388 converter_op.output_ids.push_back(-i); in GPUOperationFromNode()
36 std::vector<int> output_ids; member
37 gpu_subgraph->operations[0].output_ids.push_back(outputs[i]->id); in InitSingleOpSubgraph()
432 std::vector<int64_t> output_ids; in ExecuteProgram() local437 output_ids.push_back( in ExecuteProgram()454 for (auto output_id : output_ids) { in ExecuteProgram()
294 metal_node.outputs.resize(gpu_op.output_ids.size()); in Compile()295 for (int j = 0; j < gpu_op.output_ids.size(); ++j) { in Compile()296 int id = gpu_op.output_ids[j]; in Compile()
402 std::vector<::tflite::gpu::ValueId> output_ids;403 output_ids.reserve(outputs_.size());408 output_ids.push_back(output);