Searched refs:mutable_outputs (Results 1 – 5 of 5) sorted by relevance
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()
210 OpOutputs* mutable_outputs() { return &outputs_; } in mutable_outputs() function in tflite::flex::OpNode343 node_data->mutable_outputs()->ResetTensorHandles(); in ExecuteFlexOp()349 node_data->mutable_outputs()->GetTensorHandles()->data()), in ExecuteFlexOp()428 node_data->mutable_outputs()->InitializeGraphOutputs(output_set); in Init()
46 (*signature_def.mutable_outputs())[output].set_name(output); in BuildSignatureDef()438 TensorInfo& out = (*signature_def.mutable_outputs())["output_arg"]; in TEST_F()466 TensorInfo& out = (*signature_def.mutable_outputs())["output_arg"]; in TEST_F()
57 (*signature_def.mutable_outputs())[kSignatureOutput] = output_tensor; in GetTestSignatureDef()
1781 op_context_with_output.op_info.mutable_outputs()->Clear(); in PredictFusedConv2DBiasActivation()1782 *op_context_with_output.op_info.mutable_outputs()->Add() = output; in PredictFusedConv2DBiasActivation()2101 new_context.op_info.mutable_outputs()->Clear(); in FusedChildContext()2102 *new_context.op_info.mutable_outputs()->Add() = output; in FusedChildContext()