Home
last modified time | relevance | path

Searched refs:mutable_inputs (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/
Dgrappler_item_builder_test.cc226 (*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()
Dmutable_graph_view.cc1057 auto mutable_inputs = node->mutable_input(); in RemoveRegularFaninInternal() local
1075 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() local
1144 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()
/external/tensorflow/tensorflow/contrib/saved_model/cc/saved_model/
Dsignature_def_utils_test.cc38 (*signature_def->mutable_inputs())[key].set_name(name); in SetInputNameForKey()
106 (*signature_def_.mutable_inputs())[key].set_dtype(dtype); in SetInputDataTypeForKey()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmodel_pruner.cc349 auto mutable_inputs = node->mutable_input(); in RewriteIdentityNAndInputsOutputs() local
354 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()
/external/tensorflow/tensorflow/lite/delegates/flex/
Dkernel.cc193 OpInputs* mutable_inputs() { return &inputs_; } in mutable_inputs() function in tflite::flex::kernel::OpNode
441 node_data->mutable_inputs()->InitializeTensorSources(tflite_tensor_sources); in Init()
505 node_data->mutable_inputs()->SetForwardable(i, f); in Prepare()
/external/tensorflow/tensorflow/contrib/session_bundle/
Dbundle_shim.cc269 (*signature_def->mutable_inputs())[input_key] = tensor_info; in AddInputToSignatureDef()
/external/tensorflow/tensorflow/core/grappler/costs/
Dutils.cc120 OpInfo::TensorProperties* input = op_info->mutable_inputs(i); in ExtractExtraProperties()
Dop_level_cost_estimator.cc1411 new_context.op_info.mutable_inputs()->Clear(); in FusedChildContext()
1413 *new_context.op_info.mutable_inputs()->Add() = input; in FusedChildContext()
Dop_level_cost_estimator_test.cc386 auto* tensor_proto = op_info.mutable_inputs(0)->mutable_value(); in DescribePoolingOp()
/external/tensorflow/tensorflow/cc/tools/
Dfreeze_saved_model_test.cc43 (*signature_def.mutable_inputs())[input].set_name(input); in BuildSignatureDef()
/external/tensorflow/tensorflow/core/profiler/internal/
Dtfprof_node.h404 (*node_.mutable_inputs())[inp.first] = nodes_map.at(inp.second)->id(); in ToProto()