Searched refs:mutable_inputs (Results 1 – 10 of 10) sorted by relevance
367 auto mutable_inputs = node->mutable_input(); in RewriteIdentityNAndInputsOutputs() local372 mutable_inputs->SwapElements(i, curr_pos); in RewriteIdentityNAndInputsOutputs()380 mutable_inputs->SwapElements(i, curr_pos++); in RewriteIdentityNAndInputsOutputs()382 mutable_inputs->DeleteSubrange(curr_pos, num_inputs - curr_pos); in RewriteIdentityNAndInputsOutputs()
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()
1060 auto mutable_inputs = node->mutable_input(); in RemoveRegularFaninInternal() local1078 mutable_inputs->SwapElements(i, curr_pos); in RemoveRegularFaninInternal()1095 mutable_inputs->DeleteSubrange(curr_pos, i - curr_pos); in RemoveRegularFaninInternal()1140 auto mutable_inputs = node->mutable_input(); in RemoveRegularFaninByPort() local1147 mutable_inputs->SwapElements(i - 1, i); in RemoveRegularFaninByPort()1151 mutable_inputs->SwapElements(last_regular_fanin_port, last_node_input); in RemoveRegularFaninByPort()1153 mutable_inputs->RemoveLast(); in RemoveRegularFaninByPort()
207 OpInputs* mutable_inputs() { return &inputs_; } in mutable_inputs() function in tflite::flex::OpNode438 node_data->mutable_inputs()->InitializeTensorSources(tflite_tensor_sources); in Init()508 node_data->mutable_inputs()->SetForwardable(i, f); in Prepare()
43 (*signature_def.mutable_inputs())[input].set_name(input); in BuildSignatureDef()434 TensorInfo& in = (*signature_def.mutable_inputs())["input_arg"]; in TEST_F()461 TensorInfo& in = (*signature_def.mutable_inputs())["input_arg"]; in TEST_F()
56 (*signature_def.mutable_inputs())[kSignatureInput] = input_tensor; in GetTestSignatureDef()
121 OpInfo::TensorProperties* input = op_info->mutable_inputs(i); in ExtractExtraProperties()
1167 batch_matmul_op_info.mutable_inputs()->Clear(); in GenerateBatchMatmulContextFromEinsum()2095 new_context.op_info.mutable_inputs()->Clear(); in FusedChildContext()2097 *new_context.op_info.mutable_inputs()->Add() = input; in FusedChildContext()
446 auto* tensor_proto = op_info.mutable_inputs(0)->mutable_value(); in DescribePoolingOp()
403 (*node_.mutable_inputs())[inp.first] = nodes_map.at(inp.second)->id(); in ToProto()