Home
last modified time | relevance | path

Searched refs:arg_nodes (Results 1 – 23 of 23) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dpartitioning_utils.cc81 std::vector<std::pair<Node*, FunctionArgIndex>> arg_nodes; in UpdateArgAndRetvalMetadata() local
95 arg_nodes.emplace_back(node, FunctionArgIndex(index, sub_index)); in UpdateArgAndRetvalMetadata()
112 std::sort(arg_nodes.begin(), arg_nodes.end(), arg_comparator); in UpdateArgAndRetvalMetadata()
118 arg_indices->reserve(arg_nodes.size()); in UpdateArgAndRetvalMetadata()
119 for (const auto& pair : arg_nodes) arg_indices->push_back(pair.second); in UpdateArgAndRetvalMetadata()
123 for (int i = 0; i < arg_nodes.size(); ++i) { in UpdateArgAndRetvalMetadata()
124 Node* arg = arg_nodes[i].first; in UpdateArgAndRetvalMetadata()
Dfunction_body.cc30 this->arg_nodes.resize(arg_types.size()); in FunctionBody()
39 node_vec = &this->arg_nodes; in FunctionBody()
Dfunction_body.h38 gtl::InlinedVector<Node*, 4> arg_nodes; member
Dprocess_function_library_runtime.cc370 const std::vector<Node*>& arg_nodes) { in SetArgShape() argument
371 for (Node* n : arg_nodes) { in SetArgShape()
425 const std::vector<Node*>& arg_nodes, const std::vector<Node*>& ret_nodes, in PinArgsAndRets() argument
432 for (Node* node : arg_nodes) { in PinArgsAndRets()
635 std::vector<Node*>* arg_nodes, std::vector<Node*>* ret_nodes, in GetGraphAndArgRets() argument
647 arg_nodes->reserve(fbody->arg_nodes.size()); in GetGraphAndArgRets()
648 std::copy(fbody->arg_nodes.begin(), fbody->arg_nodes.end(), in GetGraphAndArgRets()
649 std::back_inserter(*arg_nodes)); in GetGraphAndArgRets()
712 std::vector<Node*> arg_nodes, ret_nodes; in InstantiateMultiDevice() local
718 function_name, attrs, fdef, lib_def, &graph, &arg_nodes, &ret_nodes, in InstantiateMultiDevice()
[all …]
Dinline_function_utils.cc357 num_node_inputs != fbody->arg_nodes.size()) { in ValidateInlining()
361 " arg_nodes=", fbody->arg_nodes.size()); in ValidateInlining()
690 for (std::size_t i = 0; i < fbody->arg_nodes.size(); ++i) { in InlineFunctionBody()
691 Node* arg = node_map[fbody->arg_nodes[i]->id()]; in InlineFunctionBody()
708 node_map[fbody->arg_nodes[i]->id()] = n; in InlineFunctionBody()
Dfunction.cc1426 CHECK_EQ(fbody_->arg_types.size(), fbody_->arg_nodes.size()); in Copy()
1428 for (std::size_t i = 0; i < fbody_->arg_nodes.size(); ++i) { in Copy()
1429 gbody->arg_nodes.push_back(node_map[fbody_->arg_nodes[i]->id()]); in Copy()
1461 const int index = static_cast<int>(gbody->arg_nodes.size()); in Compute()
1464 gbody->arg_nodes.push_back(dy); in Compute()
1469 const size_t num_x = fbody_->arg_nodes.size(); in Compute()
1472 for (size_t i = 0; i < fbody_->arg_nodes.size(); ++i) { in Compute()
1473 x_node_outputs.push_back({gbody->arg_nodes[i], 0}); in Compute()
Dinspecting_placer.cc154 converter.AssignGroups(fbody->arg_nodes, &groups->input_groups); in ComputeIOColocationGroups()
Dprocess_function_library_runtime.h387 const std::vector<Node*>& arg_nodes,
/external/tensorflow/tensorflow/compiler/tf2xla/
Dgraph_compiler_util.cc57 std::unordered_set<const Node*>* arg_nodes) { in AddArgNodes() argument
91 arg_nodes->insert(arg_node); in AddArgNodes()
201 Status CollectArgNodes(const Graph& graph, std::vector<Node*>* arg_nodes) { in CollectArgNodes() argument
217 arg_nodes->clear(); in CollectArgNodes()
219 const int arg_nodes_size = arg_nodes->size(); in CollectArgNodes()
223 arg_nodes->size(), ", but got index ", index_node.first); in CollectArgNodes()
225 arg_nodes->push_back(index_node.second); in CollectArgNodes()
234 std::vector<Node*> arg_nodes; in CreateXlaArgs() local
235 TF_RETURN_IF_ERROR(CollectArgNodes(graph, &arg_nodes)); in CreateXlaArgs()
236 for (const Node* node : arg_nodes) { in CreateXlaArgs()
Drearrange_function_argument.cc163 const gtl::InlinedVector<Node*, 4>* arg_nodes, // non-absl ok in RearrangeArgNodes() argument
165 for (int i = 0; i < arg_nodes->size(); i++) { in RearrangeArgNodes()
166 Node* n = (*arg_nodes)[i]; in RearrangeArgNodes()
350 RearrangeArgNodes(&fbody->arg_nodes, index_mapping); in MaybeRewriteWhileNode()
441 RearrangeArgNodes(&fbody->arg_nodes, index_mapping); in MaybeRewriteIfNode()
Dtf2xla_util.cc115 std::unordered_map<int, Node*> arg_nodes; in ReplaceArgUsageWithConstNode() local
120 arg_nodes[index] = n; in ReplaceArgUsageWithConstNode()
132 Node* arg_node = arg_nodes[arg_index]; in ReplaceArgUsageWithConstNode()
845 Node* fwd_arg = fwd_fbody->arg_nodes[fwd_while_dst_input]; in RewriteTensorListWithConstElement()
882 Node* bwd_arg = bwd_fbody->arg_nodes[bwd_while_dst_input]; in RewriteTensorListWithConstElement()
Dxla_compiler.cc755 TF_RETURN_IF_ERROR(GetNodeAttr(fbody->arg_nodes[i]->def(), "T", &dtype)); in CompileFunction()
767 fbody->arg_nodes[i]->ClearAttr("_output_shapes"); in CompileFunction()
768 fbody->arg_nodes[i]->AddAttr("_output_shapes", in CompileFunction()
773 fbody->arg_nodes[i]->ClearAttr("_output_shapes"); in CompileFunction()
774 fbody->arg_nodes[i]->AddAttr("_output_shapes", in CompileFunction()
Dconst_analysis.cc115 Node* arg_i = fbody->arg_nodes[i]; in GetCompileTimeConstInputs()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dimport_model.cc218 absl::InlinedVector<OutputTensor, 4>* arg_nodes,
230 const absl::InlinedVector<OutputTensor, 4>& arg_nodes,
351 const absl::InlinedVector<OutputTensor, 4>& arg_nodes,
1215 const FunctionBody& fbody, absl::InlinedVector<OutputTensor, 4>* arg_nodes, in GetArgsAndRetsFromFunctionBody() argument
1218 arg_nodes->reserve(fbody.arg_nodes.size()); in GetArgsAndRetsFromFunctionBody()
1220 for (auto arg : fbody.arg_nodes) { in GetArgsAndRetsFromFunctionBody()
1221 arg_nodes->emplace_back(arg, 0); in GetArgsAndRetsFromFunctionBody()
1317 absl::InlinedVector<OutputTensor, 4> arg_nodes; in ConvertLibFunction() local
1320 GetArgsAndRetsFromFunctionBody(*fbody, &arg_nodes, &ret_nodes, in ConvertLibFunction()
1324 mlir_func_name, func_type, arg_nodes, ret_nodes, control_ret_nodes, in ConvertLibFunction()
[all …]
/external/tensorflow/tensorflow/core/framework/
Dgraph_to_functiondef.h66 std::vector<Node*>* arg_nodes, std::vector<Node*>* ret_nodes,
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dwhile_op.cc54 body->arg_nodes[i]->name(), " is not a resource."); in VerifyResourceArgsGroupedAtEnd()
123 const Node* arg = body->arg_nodes[i]; in GetLoopInvariants()
156 (body_must_be_const_nodes[body->arg_nodes[arg_idx]->id()] || in ConvertLoopInvariantsToConst()
157 cond_must_be_const_nodes[cond->arg_nodes[arg_idx]->id()]); in ConvertLoopInvariantsToConst()
Dif_op.cc226 (then_branch_must_be_const_nodes[then_body->arg_nodes[arg_idx] in Compile()
228 else_branch_must_be_const_nodes[else_body->arg_nodes[arg_idx] in Compile()
Dcase_op.cc143 [case_bodies[branch_idx]->arg_nodes[arg_idx]->id()]) { in Compile()
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dvectorization_utils.cc519 TF_RETURN_IF_ERROR(add_conversion(map_defun_fn_->arg_nodes[i], true)); in AddArgTensorMappings()
524 for (int i = num_args, end = map_defun_fn_->arg_nodes.size(); i < end; ++i) { in AddArgTensorMappings()
525 TF_RETURN_IF_ERROR(add_conversion(map_defun_fn_->arg_nodes[i], false)); in AddArgTensorMappings()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dencapsulate_tpu_computations_pass.cc1726 const Node* then_arg_node = then_branch_fbody->arg_nodes[i]; in FindArgsToLiftForIfNode()
1737 const Node* else_arg_node = else_branch_fbody->arg_nodes[i]; in FindArgsToLiftForIfNode()
1786 for (int i = 0; i < cond_fbody->arg_nodes.size(); i++) { in FindArgsToLiftForWhileNode()
1787 const Node* arg_node = cond_fbody->arg_nodes[i]; in FindArgsToLiftForWhileNode()
1811 if (node != body_fbody->arg_nodes[i]) { in FindArgsToLiftForWhileNode()
1818 for (int i = 0; i < body_fbody->arg_nodes.size(); i++) { in FindArgsToLiftForWhileNode()
1819 const Node* arg_node = body_fbody->arg_nodes[i]; in FindArgsToLiftForWhileNode()
1829 for (int i = 0; i < body_fbody->arg_nodes.size(); i++) { in FindArgsToLiftForWhileNode()
1830 const Node* arg_node = body_fbody->arg_nodes[i]; in FindArgsToLiftForWhileNode()
1859 for (int i = 0; i < fbody.arg_nodes.size(); i++) { in FindArgsToLiftForCallNode()
[all …]
/external/tensorflow/tensorflow/compiler/jit/
Dextract_outside_compilation_pass.cc88 Status GetArgDataTypes(const std::vector<Node*>& arg_nodes, in GetArgDataTypes() argument
90 recv_at_host_dtypes->resize(arg_nodes.size(), DT_INVALID); in GetArgDataTypes()
91 for (auto* n : arg_nodes) { in GetArgDataTypes()
137 std::vector<Node*> arg_nodes = GatherNodesWithType(*g, "_Arg"); in ReplaceArgNodesWithRecvAtHostNode() local
138 TF_RETURN_IF_ERROR(GetArgDataTypes(arg_nodes, recv_at_host_dtypes)); in ReplaceArgNodesWithRecvAtHostNode()
143 for (auto* n : arg_nodes) { in ReplaceArgNodesWithRecvAtHostNode()
605 int original_arg_count = body_function_body->arg_nodes.size(); in PostprocessLiftedArgsForWhile()
723 int original_arg_count = then_branch_function_body->arg_nodes.size(); in PostprocessLiftedArgsForIf()
837 int original_arg_count = fbody->arg_nodes.size(); in PostprocessLiftedArgsForCall()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_compile_op_common.cc330 TF_RET_CHECK(fbody->arg_nodes.size() == args.size()); in CompileTFFunctionToHlo()
331 for (size_t i = 0; i < fbody->arg_nodes.size(); i++) { in CompileTFFunctionToHlo()
332 args[i].node_name = fbody->arg_nodes[i]->name(); in CompileTFFunctionToHlo()
/external/tensorflow/tensorflow/core/grappler/utils/
Dfunctions.cc266 const Node* node = fbody->arg_nodes[in_id]; in MakeGrapplerFunctionItem()