Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/core/kernels/
Drestore_v2_op_test.cc216 (*mutable_input(1).tensor).flat<string>()(0) = tensor_names[1]; in RunTest()
228 (*mutable_input(1).tensor).flat<string>()(0) = tensor_names[2]; in RunTest()
240 (*mutable_input(1).tensor).flat<string>()(0) = tensor_names[3]; in RunTest()
252 (*mutable_input(1).tensor).flat<string>()(0) = tensor_names[4]; in RunTest()
264 (*mutable_input(1).tensor).flat<string>()(0) = tensor_names[5]; in RunTest()
277 (*mutable_input(1).tensor).flat<string>()(0) = tensor_names[6]; in RunTest()
289 (*mutable_input(1).tensor).flat<string>()(0) = tensor_names[7]; in RunTest()
301 (*mutable_input(1).tensor).flat<string>()(0) = tensor_names[8]; in RunTest()
313 (*mutable_input(1).tensor).flat<string>()(0) = tensor_names[9]; in RunTest()
325 (*mutable_input(1).tensor).flat<string>()(0) = tensor_names[10]; in RunTest()
[all …]
Drestore_op_test.cc206 (*mutable_input(1).tensor).scalar<string>()() = tensor_names[1]; in TEST_F()
218 (*mutable_input(1).tensor).scalar<string>()() = tensor_names[2]; in TEST_F()
230 (*mutable_input(1).tensor).scalar<string>()() = tensor_names[3]; in TEST_F()
242 (*mutable_input(1).tensor).scalar<string>()() = tensor_names[4]; in TEST_F()
254 (*mutable_input(1).tensor).scalar<string>()() = tensor_names[5]; in TEST_F()
267 (*mutable_input(1).tensor).scalar<string>()() = tensor_names[6]; in TEST_F()
279 (*mutable_input(1).tensor).scalar<string>()() = tensor_names[7]; in TEST_F()
291 (*mutable_input(1).tensor).scalar<string>()() = tensor_names[8]; in TEST_F()
303 (*mutable_input(1).tensor).scalar<string>()() = tensor_names[9]; in TEST_F()
315 (*mutable_input(1).tensor).scalar<string>()() = tensor_names[10]; in TEST_F()
[all …]
Dscatter_op_test.cc58 Tensor params_tensor = *mutable_input(0).tensor; in TEST_F()
71 Tensor params_tensor = *mutable_input(0).tensor; in TEST_F()
89 Tensor params_tensor = *mutable_input(0).tensor; in TEST_F()
108 Tensor params_tensor = *mutable_input(0).tensor; in TEST_F()
125 Tensor params_tensor = *mutable_input(0).tensor; in TEST_F()
141 Tensor params_tensor = *mutable_input(0).tensor; in TEST_F()
157 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()
Dgpu_utils.cc65 input.shape().AsProto(instr.mutable_input()->mutable_tensor_shape()); in LogConvAutotuneResults()
66 instr.mutable_input()->set_dtype(input.dtype()); in LogConvAutotuneResults()
88 input.shape().AsProto(instr.mutable_input()->mutable_tensor_shape()); in LogFusedConvAutotuneResults()
89 instr.mutable_input()->set_dtype(input.dtype()); in LogFusedConvAutotuneResults()
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()
167 const Tensor& input_tensor = context->mutable_input(0, false); in Compute()
Dcount_up_to_op.cc35 Tensor tensor = context->mutable_input(0, true); in Compute()
/external/tensorflow/tensorflow/contrib/session_bundle/
Dsignature_test.cc49 input_signature->mutable_input()->set_tensor_name("flow"); in TEST()
101 input_signature->mutable_input()->set_tensor_name("flow"); in TEST()
156 input_signature->mutable_input()->set_tensor_name("flow"); in TEST()
208 input_signature->mutable_input()->set_tensor_name("flow"); in TEST()
288 signature_.mutable_input()->set_tensor_name(kInputName); in SetUp()
365 signature.mutable_input()->set_tensor_name("in:0"); in TEST()
380 signature.mutable_input()->set_tensor_name("in:0"); in TEST()
397 signature.mutable_input()->set_tensor_name("in:0"); in TEST()
418 signature_.mutable_input()->set_tensor_name(kInputName); in SetUp()
477 ->mutable_input() in TEST()
[all …]
Dbundle_shim_test.cc173 regression_signature->mutable_input()->set_tensor_name("foo-input"); in TEST()
200 classification_signature->mutable_input()->set_tensor_name("foo-input"); in TEST()
253 foo_regression_signature->mutable_input()->set_tensor_name("foo-input"); in TEST()
259 bar_regression_signature->mutable_input()->set_tensor_name("bar-input"); in TEST()
306 foo_classification_signature->mutable_input()->set_tensor_name("foo-input"); in TEST()
313 bar_classification_signature->mutable_input()->set_tensor_name("bar-input"); in TEST()
471 inputs_regression_signature->mutable_input()->set_tensor_name("bar-input"); in TEST()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Ddependency_optimizer.cc46 node->mutable_input()->SwapElements(pos, node->input_size() - 1); in RemoveInput()
47 node->mutable_input()->RemoveLast(); in RemoveInput()
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()
568 target_node->mutable_input()->SwapElements(input_slot, in TransitiveReduction()
571 target_node->mutable_input()->RemoveLast(); in TransitiveReduction()
655 node->mutable_input()->SwapElements(pos, node->input_size() - 1); in GroupCrossDeviceControlEdges()
656 node->mutable_input()->RemoveLast(); in GroupCrossDeviceControlEdges()
Ddebug_stripper.cc39 for (string& inp : *node.mutable_input()) { in Optimize()
57 *node.mutable_input(i) = AsControlDependency(NodeName(node.input(i))); in Optimize()
Dloop_optimizer.cc149 *consumer->mutable_input(i) = AsControlDependency(*const_node); in HandleConst()
151 *consumer->mutable_input(i) = const_node->name(); in HandleConst()
196 node->mutable_input()->SwapElements(i, node->input_size() - 1); in HandleInvariantNode()
197 node->mutable_input()->RemoveLast(); in HandleInvariantNode()
249 *consumer->mutable_input(i) = new_enter->name(); in HandleInvariantNode()
539 push_node->mutable_input()->SwapElements(0, 1); in RemoveStackOps()
876 dead_node->mutable_input()->DeleteSubrange(index, 1); in RemoveDeadBranches()
889 auto* inputs = dead_node->mutable_input(); in RemoveDeadBranches()
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()
Dconstant_folding.cc129 node->mutable_input()->SwapElements(i, node->input_size() - 1); in MaybeRemoveControlInput()
130 node->mutable_input()->RemoveLast(); in MaybeRemoveControlInput()
490 *output->mutable_input(k) = const_name; in MaterializeShapes()
661 *output->mutable_input(k) = out[port]->name(); in MaterializeBroadcastGradientArgs()
1300 *output->mutable_input(i) = const_out->name(); in FoldMergeNode()
1303 *output->mutable_input(i) = const_index->name(); in FoldMergeNode()
1368 *node->mutable_input() = const_node->input(); in FoldNode()
1407 *output->mutable_input(i) = AsControlDependency(*constant_output); in FoldNode()
1414 *output->mutable_input(i) = const_nodes[port].name(); in FoldNode()
1651 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.cc209 n->mutable_input()->SwapElements(i, i + 1); in RemoveInputAtIndex()
211 n->mutable_input()->RemoveLast(); in RemoveInputAtIndex()
532 replaced_graph_def.mutable_node(i)->mutable_input()->RemoveLast(); in SparsifyGatherInternal()
/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()
870 node->mutable_input()->SwapElements(last_node_input, num_regular_fanins); in AddFaninInternal()
931 node->mutable_input()->SwapElements(num_regular_fanins, last_node_input); in AddRegularFaninByPort()
938 node->mutable_input()->SwapElements(i, i + 1); in AddRegularFaninByPort()
1057 auto mutable_inputs = node->mutable_input(); in RemoveRegularFaninInternal()
1137 auto mutable_inputs = node->mutable_input(); in RemoveRegularFaninByPort()
1172 node->mutable_input()->SwapElements(i, node->input_size() - 1); in RemoveControllingFaninInternal()
1173 node->mutable_input()->RemoveLast(); in RemoveControllingFaninInternal()
1222 node->mutable_input()->DeleteSubrange(0, num_regular_fanins); in RemoveAllFanins()
[all …]
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
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/contrib/input_pipeline/kernels/
Dinput_pipeline_kernels.cc46 OP_REQUIRES_OK(ctx, ctx->mutable_input("counter", &counter_tensor, true)); in Compute()
/external/tensorflow/tensorflow/core/grappler/utils/
Dgrappler_test.cc56 auto want_inputs = want_node.mutable_input(); in CompareGraphNodes()
57 auto got_inputs = got_node.mutable_input(); in CompareGraphNodes()
/external/tensorflow/tensorflow/core/framework/
Dgraph_to_functiondef.cc243 *node_def->mutable_input(i) = strings::StrCat("^", normalized); in GraphToFunctionDef()
251 *node_def->mutable_input(i) = iter->second; in GraphToFunctionDef()
/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/
Dscatter_add_ndim_op.cc34 Tensor input_tensor = context->mutable_input(0, false); in Compute()

123