Searched refs:mutable_inputs (Results 1 – 11 of 11) sorted by relevance
226 (*serving_signature.mutable_inputs())["input"] = input; in TEST_F()236 (*serving_signature.mutable_inputs())["input2"] = input2; in TEST_F()268 (*serving_signature.mutable_inputs())["input"] = input; in TEST_F()295 (*serving_signature.mutable_inputs())["input"] = input; in TEST_F()
1057 auto mutable_inputs = node->mutable_input(); in RemoveRegularFaninInternal() local1075 mutable_inputs->SwapElements(i, curr_pos); in RemoveRegularFaninInternal()1092 mutable_inputs->DeleteSubrange(curr_pos, i - curr_pos); in RemoveRegularFaninInternal()1137 auto mutable_inputs = node->mutable_input(); in RemoveRegularFaninByPort() local1144 mutable_inputs->SwapElements(i - 1, i); in RemoveRegularFaninByPort()1148 mutable_inputs->SwapElements(last_regular_fanin_port, last_node_input); in RemoveRegularFaninByPort()1150 mutable_inputs->RemoveLast(); in RemoveRegularFaninByPort()
38 (*signature_def->mutable_inputs())[key].set_name(name); in SetInputNameForKey()106 (*signature_def_.mutable_inputs())[key].set_dtype(dtype); in SetInputDataTypeForKey()
349 auto mutable_inputs = node->mutable_input(); in RewriteIdentityNAndInputsOutputs() local354 mutable_inputs->SwapElements(i, curr_pos); in RewriteIdentityNAndInputsOutputs()362 mutable_inputs->SwapElements(i, curr_pos++); in RewriteIdentityNAndInputsOutputs()364 mutable_inputs->DeleteSubrange(curr_pos, num_inputs - curr_pos); in RewriteIdentityNAndInputsOutputs()
193 OpInputs* mutable_inputs() { return &inputs_; } in mutable_inputs() function in tflite::flex::kernel::OpNode441 node_data->mutable_inputs()->InitializeTensorSources(tflite_tensor_sources); in Init()505 node_data->mutable_inputs()->SetForwardable(i, f); in Prepare()
269 (*signature_def->mutable_inputs())[input_key] = tensor_info; in AddInputToSignatureDef()
120 OpInfo::TensorProperties* input = op_info->mutable_inputs(i); in ExtractExtraProperties()
1411 new_context.op_info.mutable_inputs()->Clear(); in FusedChildContext()1413 *new_context.op_info.mutable_inputs()->Add() = input; in FusedChildContext()
386 auto* tensor_proto = op_info.mutable_inputs(0)->mutable_value(); in DescribePoolingOp()
43 (*signature_def.mutable_inputs())[input].set_name(input); in BuildSignatureDef()
404 (*node_.mutable_inputs())[inp.first] = nodes_map.at(inp.second)->id(); in ToProto()