Searched refs:mutable_outputs (Results 1 – 6 of 6) sorted by relevance
43 (*signature_def->mutable_outputs())[key].set_name(name); in SetOutputNameForKey()110 (*signature_def_.mutable_outputs())[key].set_dtype(dtype); in SetOutputDataTypeForKey()114 (*signature_def_.mutable_outputs()).erase(key); in EraseOutputKey()
227 (*serving_signature.mutable_outputs())["output"] = output; in TEST_F()237 (*serving_signature.mutable_outputs())["output2"] = output2; in TEST_F()269 (*serving_signature.mutable_outputs())["output"] = output; in TEST_F()296 (*serving_signature.mutable_outputs())["output"] = output; in TEST_F()
196 OpOutputs* mutable_outputs() { return &outputs_; } in mutable_outputs() function in tflite::flex::kernel::OpNode340 node_data->mutable_outputs()->ResetTensorHandles(); in ExecuteFlexOp()344 node_data->mutable_outputs()->GetTensorHandles(), in ExecuteFlexOp()431 node_data->mutable_outputs()->InitializeGraphOutputs(output_set); in Init()
299 (*signature_def->mutable_outputs())[output_key] = tensor_info; in AddOutputToSignatureDef()
1214 op_context_with_output.op_info.mutable_outputs()->Clear(); in PredictFusedConv2DBiasActivation()1215 *op_context_with_output.op_info.mutable_outputs()->Add() = output; in PredictFusedConv2DBiasActivation()1417 new_context.op_info.mutable_outputs()->Clear(); in FusedChildContext()1418 *new_context.op_info.mutable_outputs()->Add() = output; in FusedChildContext()
46 (*signature_def.mutable_outputs())[output].set_name(output); in BuildSignatureDef()