Home
last modified time | relevance | path

Searched refs:mutable_input (Results 1 – 25 of 59) sorted by relevance

123

/external/tensorflow/tensorflow/core/kernels/
Drestore_v2_op_test.cc219 (*mutable_input(1).tensor).flat<tstring>()(0) = tensor_names[1]; in RunTest()
231 (*mutable_input(1).tensor).flat<tstring>()(0) = tensor_names[2]; in RunTest()
243 (*mutable_input(1).tensor).flat<tstring>()(0) = tensor_names[3]; in RunTest()
255 (*mutable_input(1).tensor).flat<tstring>()(0) = tensor_names[4]; in RunTest()
267 (*mutable_input(1).tensor).flat<tstring>()(0) = tensor_names[5]; in RunTest()
280 (*mutable_input(1).tensor).flat<tstring>()(0) = tensor_names[6]; in RunTest()
292 (*mutable_input(1).tensor).flat<tstring>()(0) = tensor_names[7]; in RunTest()
304 (*mutable_input(1).tensor).flat<tstring>()(0) = tensor_names[8]; in RunTest()
316 (*mutable_input(1).tensor).flat<tstring>()(0) = tensor_names[9]; in RunTest()
328 (*mutable_input(1).tensor).flat<tstring>()(0) = tensor_names[10]; in RunTest()
[all …]
Drestore_op_test.cc209 (*mutable_input(1).tensor).scalar<tstring>()() = tensor_names[1]; in TEST_F()
221 (*mutable_input(1).tensor).scalar<tstring>()() = tensor_names[2]; in TEST_F()
233 (*mutable_input(1).tensor).scalar<tstring>()() = tensor_names[3]; in TEST_F()
245 (*mutable_input(1).tensor).scalar<tstring>()() = tensor_names[4]; in TEST_F()
257 (*mutable_input(1).tensor).scalar<tstring>()() = tensor_names[5]; in TEST_F()
270 (*mutable_input(1).tensor).scalar<tstring>()() = tensor_names[6]; in TEST_F()
282 (*mutable_input(1).tensor).scalar<tstring>()() = tensor_names[7]; in TEST_F()
294 (*mutable_input(1).tensor).scalar<tstring>()() = tensor_names[8]; in TEST_F()
306 (*mutable_input(1).tensor).scalar<tstring>()() = tensor_names[9]; in TEST_F()
318 (*mutable_input(1).tensor).scalar<tstring>()() = tensor_names[10]; in TEST_F()
[all …]
Dscatter_op_test.cc69 Tensor params_tensor = *mutable_input(0).tensor; in TEST_F()
82 Tensor params_tensor = *mutable_input(0).tensor; in TEST_F()
100 Tensor params_tensor = *mutable_input(0).tensor; in TEST_F()
119 Tensor params_tensor = *mutable_input(0).tensor; in TEST_F()
136 Tensor params_tensor = *mutable_input(0).tensor; in TEST_F()
152 Tensor params_tensor = *mutable_input(0).tensor; in TEST_F()
168 Tensor params_tensor = *mutable_input(0).tensor; in TEST_F()
214 Tensor params_tensor = *mutable_input(0).tensor; in TEST_F()
Dscatter_nd_op_test.cc90 Tensor params_tensor = *mutable_input(0).tensor; in TEST_F()
109 Tensor params_tensor = *mutable_input(0).tensor; in TEST_F()
139 Tensor params_tensor = *mutable_input(0).tensor; in TEST_F()
155 Tensor params_tensor = *mutable_input(0).tensor; in TEST_F()
171 Tensor params_tensor = *mutable_input(0).tensor; in TEST_F()
Dassign_op.h65 const Tensor& old_lhs = context->mutable_input(0, /* lock_held */ true); in Compute()
132 Tensor old_unlocked_lhs = context->mutable_input(0, /* lock_held */ false); in Compute()
Dvariable_ops.cc144 Tensor tmpvar = context->mutable_input(0, false); in Compute()
168 const Tensor& input_tensor = context->mutable_input(0, false); in Compute()
Dgpu_utils.cc141 *instr.mutable_input() = input_desc.ToProto(element_type); in LogConvAutotuneResults()
183 *instr.mutable_input() = input_desc.ToProto(element_type); in LogFusedConvForwardAutotuneResults()
Dcount_up_to_op.cc36 Tensor tensor = context->mutable_input(0, true); in Compute()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Ddependency_optimizer.cc48 node->mutable_input()->SwapElements(pos, node->input_size() - 1); in RemoveControlInput()
49 node->mutable_input()->RemoveLast(); in RemoveControlInput()
244 fanout->mutable_input()->SwapElements(i, fanout->input_size() - 1); in OptimizeNode()
245 fanout->mutable_input()->RemoveLast(); in OptimizeNode()
280 node->mutable_input()->SwapElements(pos, node->input_size() - 1); in OptimizeNode()
281 node->mutable_input()->RemoveLast(); in OptimizeNode()
589 target_node->mutable_input()->SwapElements(input_slot, in TransitiveReduction()
592 target_node->mutable_input()->RemoveLast(); in TransitiveReduction()
699 node->mutable_input()->SwapElements(pos, node->input_size() - 1); in GroupCrossDeviceControlEdges()
700 node->mutable_input()->RemoveLast(); in GroupCrossDeviceControlEdges()
Ddebug_stripper.cc50 for (string& inp : *node.mutable_input()) { in Optimize()
68 *node.mutable_input(i) = AsControlDependency(NodeName(node.input(i))); in Optimize()
Dauto_parallel.cc145 *apply_gradients_node->mutable_input(gradient_pos[apply_gradients_op]) = in Initialize()
212 *new_node->mutable_input(i) = new_name; in AddSharedNodes()
231 *new_node->mutable_input(i) = new_name; in AddOneReplica()
Dloop_optimizer.cc148 *consumer->mutable_input(i) = AsControlDependency(*const_node); in HandleConst()
150 *consumer->mutable_input(i) = const_node->name(); in HandleConst()
194 node->mutable_input()->SwapElements(i, node->input_size() - 1); in HandleInvariantNode()
195 node->mutable_input()->RemoveLast(); in HandleInvariantNode()
247 *consumer->mutable_input(i) = new_enter->name(); in HandleInvariantNode()
538 push_node->mutable_input()->SwapElements(0, 1); in RemoveStackOps()
932 auto* inputs = merge_node->mutable_input(); in RemoveDeadBranches()
Dconstant_folding.cc112 node->mutable_input()->SwapElements(i, node->input_size() - 1); in MaybeRemoveControlInput()
113 node->mutable_input()->RemoveLast(); in MaybeRemoveControlInput()
520 *output->mutable_input(k) = const_name; in MaterializeShapes()
691 *output->mutable_input(k) = out[port]->name(); in MaterializeBroadcastGradientArgs()
1404 *output->mutable_input(i) = const_out->name(); in FoldMergeNode()
1407 *output->mutable_input(i) = const_index->name(); in FoldMergeNode()
1480 *node->mutable_input() = const_node->input(); in FoldNode()
1519 *output->mutable_input(i) = AsControlDependency(*constant_output); in FoldNode()
1526 *output->mutable_input(i) = const_nodes[port].name(); in FoldNode()
1766 node->mutable_input()->SwapElements(0, input_to_forward); in ReplaceOperationWithIdentity()
[all …]
/external/tensorflow/tensorflow/tools/graph_transforms/
Dtransform_utils_test.cc447 *(identity_node.mutable_input()->Add()) = original_copy.name(); in TestReplaceMatchingOpTypes()
708 *(mul_node->mutable_input()->Add()) = "add_node1"; in TestFindInvalidInputs()
709 *(mul_node->mutable_input()->Add()) = "add_node2:0"; in TestFindInvalidInputs()
710 *(mul_node->mutable_input()->Add()) = "^const_node1:0"; in TestFindInvalidInputs()
715 *(add_node1->mutable_input()->Add()) = "missing_input1"; in TestFindInvalidInputs()
716 *(add_node1->mutable_input()->Add()) = "const_node1:0"; in TestFindInvalidInputs()
717 *(add_node1->mutable_input()->Add()) = "missing_input2"; in TestFindInvalidInputs()
722 *(add_node2->mutable_input()->Add()) = "missing_input3"; in TestFindInvalidInputs()
723 *(add_node2->mutable_input()->Add()) = "const_node1:0"; in TestFindInvalidInputs()
724 *(add_node2->mutable_input()->Add()) = "^const_node2"; in TestFindInvalidInputs()
[all …]
Dobfuscate_names.cc78 node->mutable_input()->Clear(); in ObfuscateNames()
89 *(node->mutable_input()->Add()) = new_input_name; in ObfuscateNames()
Dsparsify_gather.cc208 n->mutable_input()->SwapElements(i, i + 1); in RemoveInputAtIndex()
210 n->mutable_input()->RemoveLast(); in RemoveInputAtIndex()
531 replaced_graph_def.mutable_node(i)->mutable_input()->RemoveLast(); in SparsifyGatherInternal()
Dbackports.cc44 concat_node.mutable_input()->Clear(); in BackportConcatV2Transform()
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dmake_stateless.cc51 node.mutable_input()->RemoveLast(); in OptimizeAndCollectStats()
61 node.mutable_input()->RemoveLast(); in OptimizeAndCollectStats()
Dinject_prefetch.cc76 *prefetch_node.mutable_input()->Add() = async_dataset_node->name(); in OptimizeAndCollectStats()
78 *prefetch_node.mutable_input()->Add() = autotune_value->name(); in OptimizeAndCollectStats()
Dfunction_utils.cc78 std::replace(n.mutable_input()->begin(), n.mutable_input()->end(), from, in ReplaceReferences()
Dfusion_utils.cc135 for (string& input : *function_node.mutable_input()) { in RenameFunctionNodes()
195 for (auto& node_input : *function_node.mutable_input()) { in GetUniqueSignature()
216 for (auto& node_input : *function_node.mutable_input()) { in FuseFunctionNodes()
/external/tensorflow/tensorflow/core/grappler/utils/
Dgraph_view.cc253 node.mutable_input()->Clear(); in AddNode()
572 node->mutable_input()->SwapElements(pos, last_pos); in AddFaninsInternal()
597 node->mutable_input()->DeleteSubrange(last_pos + 1, last_idx - last_pos); in AddFaninsInternal()
955 node_def->mutable_input()->Clear(); in AddNewNodes()
1132 auto* mutable_input = node_view->node()->mutable_input(); in RemoveControllingFaninInternal() local
1147 mutable_input->SwapElements( in RemoveControllingFaninInternal()
1152 mutable_input->RemoveLast(); in RemoveControllingFaninInternal()
1208 node_def->mutable_input()->DeleteSubrange( in ApplyNodeUpdates()
1212 node_def->mutable_input()->Reserve(node_def->mutable_input()->size() + in ApplyNodeUpdates()
1219 node_def->mutable_input()->SwapElements(curr_index, in ApplyNodeUpdates()
[all …]
Dcanonicalizer.cc36 auto* input = node->mutable_input(); in CanonicalizeNode()
Dgrappler_test.cc59 auto want_inputs = want_node.mutable_input(); in CompareGraphNodes()
60 auto got_inputs = got_node.mutable_input(); in CompareGraphNodes()
/external/tensorflow/tensorflow/core/grappler/
Dmutable_graph_view.cc359 node->mutable_input()->SwapElements(pos, last_pos); in AddAndDedupFanouts()
380 node->mutable_input()->DeleteSubrange(last_pos + 1, last_idx - last_pos); in AddAndDedupFanouts()
873 node->mutable_input()->SwapElements(last_node_input, num_regular_fanins); in AddFaninInternal()
934 node->mutable_input()->SwapElements(num_regular_fanins, last_node_input); in AddRegularFaninByPort()
941 node->mutable_input()->SwapElements(i, i + 1); in AddRegularFaninByPort()
1060 auto mutable_inputs = node->mutable_input(); in RemoveRegularFaninInternal()
1140 auto mutable_inputs = node->mutable_input(); in RemoveRegularFaninByPort()
1175 node->mutable_input()->SwapElements(i, node->input_size() - 1); in RemoveControllingFaninInternal()
1176 node->mutable_input()->RemoveLast(); in RemoveControllingFaninInternal()
1225 node->mutable_input()->DeleteSubrange(0, num_regular_fanins); in RemoveAllFanins()
[all …]

123