Home
last modified time | relevance | path

Searched refs:mutable_input (Results 1 – 25 of 67) 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.cc159 Tensor tmpvar = context->mutable_input(0, false); in Compute()
184 const Tensor& input_tensor = context->mutable_input(0, false); in Compute()
Dgpu_utils.cc141 *instr.mutable_input() = input_desc.ToProto(element_type); in LogConvAutotuneResults()
184 *instr.mutable_input() = input_desc.ToProto(element_type); in LogFusedConvForwardAutotuneResults()
/external/tensorflow/tensorflow/tools/graph_transforms/
Dinline_partitionedcall.cc115 *new_node->mutable_input(), new_node->mutable_input()->begin(), in InlinePartitionedCall()
136 *node.mutable_input(), node.mutable_input()->begin(), in InlinePartitionedCall()
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.cc77 node->mutable_input()->Clear(); in ObfuscateNames()
88 *(node->mutable_input()->Add()) = new_input_name; in ObfuscateNames()
Drename_node.cc53 for (std::string& input_name : *node->mutable_input()) { in RenameNode()
/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()
266 fanout->mutable_input()->SwapElements(i, fanout->input_size() - 1); in OptimizeNode()
267 fanout->mutable_input()->RemoveLast(); in OptimizeNode()
302 node->mutable_input()->SwapElements(pos, node->input_size() - 1); in OptimizeNode()
303 node->mutable_input()->RemoveLast(); in OptimizeNode()
611 target_node->mutable_input()->SwapElements(input_slot, in TransitiveReduction()
614 target_node->mutable_input()->RemoveLast(); in TransitiveReduction()
721 node->mutable_input()->SwapElements(pos, node->input_size() - 1); in GroupCrossDeviceControlEdges()
722 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.cc146 *apply_gradients_node->mutable_input(gradient_pos[apply_gradients_op]) = in Initialize()
215 *new_node->mutable_input(i) = new_name; in AddSharedNodes()
234 *new_node->mutable_input(i) = new_name; in AddOneReplica()
Dconstant_folding.cc112 node->mutable_input()->SwapElements(i, node->input_size() - 1); in MaybeRemoveControlInput()
113 node->mutable_input()->RemoveLast(); in MaybeRemoveControlInput()
543 *output->mutable_input(k) = const_name; in MaterializeShapes()
716 *output->mutable_input(k) = out[port]->name(); in MaterializeBroadcastGradientArgs()
1463 *output->mutable_input(i) = const_out->name(); in FoldMergeNode()
1466 *output->mutable_input(i) = const_index->name(); in FoldMergeNode()
1539 *node->mutable_input() = const_node->input(); in FoldNode()
1579 *output->mutable_input(i) = AsControlDependency(*constant_output); in FoldNode()
1586 *output->mutable_input(i) = const_nodes[port].name(); in FoldNode()
1858 node->mutable_input()->SwapElements(0, input_to_broadcast); in ReplaceOperationWithBroadcastTo()
[all …]
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()
Dimplementation_selector.cc178 node_def->mutable_input()->RemoveLast(); in UpdateNodeDef()
185 for (int i = 0; i < diff; ++i) node_def->mutable_input()->RemoveLast(); in UpdateNodeDef()
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Ddisable_intra_op_parallelism.cc93 *insert_node.mutable_input()->Add() = last_node->name(); in OptimizeAndCollectStats()
95 *insert_node.mutable_input()->Add() = max_parallelism_value->name(); in OptimizeAndCollectStats()
Dreorder_data_discarding_ops.cc115 *discard_node->mutable_input(0) = start_parent->name(); in OptimizeAndCollectStats()
116 *start->mutable_input(0) = discard_node->name(); in OptimizeAndCollectStats()
Dautotune_buffer_sizes.cc121 *prefetch_node.mutable_input()->Add() = async_dataset_node->name(); in OptimizeAndCollectStats()
123 *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()
/external/tensorflow/tensorflow/core/grappler/utils/
Dgraph_view.cc254 node.mutable_input()->Clear(); in AddNode()
574 node->mutable_input()->SwapElements(pos, last_pos); in AddFaninsInternal()
601 node->mutable_input()->DeleteSubrange(last_pos + 1, last_idx - last_pos); in AddFaninsInternal()
961 node_def->mutable_input()->Clear(); in AddNewNodes()
1147 auto* mutable_input = node_view->node()->mutable_input(); in RemoveControllingFaninInternal() local
1164 mutable_input->SwapElements( in RemoveControllingFaninInternal()
1169 mutable_input->RemoveLast(); in RemoveControllingFaninInternal()
1225 node_def->mutable_input()->DeleteSubrange( in ApplyNodeUpdates()
1229 node_def->mutable_input()->Reserve(node_def->mutable_input()->size() + in ApplyNodeUpdates()
1236 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