Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla.cc208 Status CollectArgNodes(const Graph& graph, std::vector<Node*>* arg_nodes) { in CollectArgNodes() argument
223 arg_nodes->clear(); in CollectArgNodes()
225 if (index_node.first != arg_nodes->size()) { in CollectArgNodes()
227 arg_nodes->size(), ", but got index ", in CollectArgNodes()
230 arg_nodes->push_back(index_node.second); in CollectArgNodes()
238 std::vector<Node*> arg_nodes; in CreateXlaArgs() local
239 TF_RETURN_IF_ERROR(CollectArgNodes(graph, &arg_nodes)); in CreateXlaArgs()
240 for (const Node* node : arg_nodes) { in CreateXlaArgs()
Dconst_analysis.cc173 Node* arg_i = fbody->arg_nodes[i]; in GetCompileTimeConstInputs()
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()
/external/tensorflow/tensorflow/core/common_runtime/
Dpartitioning_utils.cc77 std::vector<std::pair<Node*, int>> arg_nodes; in UpdateArgAndRetvalMetadata() local
89 arg_nodes.push_back(std::make_pair(node, index)); in UpdateArgAndRetvalMetadata()
98 for (int i = 0; i < arg_nodes.size(); ++i) { in UpdateArgAndRetvalMetadata()
99 Node* arg = arg_nodes[i].first; in UpdateArgAndRetvalMetadata()
Dfunction.cc1469 num_node_inputs != fbody->arg_nodes.size()) { in ValidateInlining()
1473 " arg_nodes=", fbody->arg_nodes.size()); in ValidateInlining()
1705 for (std::size_t i = 0; i < fbody->arg_nodes.size(); ++i) { in InlineFunctionBody()
1706 Node* arg = node_map[fbody->arg_nodes[i]->id()]; in InlineFunctionBody()
1718 node_map[fbody->arg_nodes[i]->id()] = n; in InlineFunctionBody()
1926 this->arg_nodes.resize(arg_types.size()); in FunctionBody()
1933 node_vec = &this->arg_nodes; in FunctionBody()
1998 CHECK_EQ(fbody_->arg_types.size(), fbody_->arg_nodes.size()); in Copy()
2000 for (std::size_t i = 0; i < fbody_->arg_nodes.size(); ++i) { in Copy()
2001 gbody_->arg_nodes.push_back(node_map[fbody_->arg_nodes[i]->id()]); in Copy()
[all …]
Dfunction.h80 gtl::InlinedVector<Node*, 4> arg_nodes; member
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dvectorization_utils.cc526 TF_RETURN_IF_ERROR(add_conversion(map_defun_fn_->arg_nodes[i], true)); in AddArgTensorMappings()
531 for (int i = num_args; i < map_defun_fn_->arg_nodes.size(); ++i) { in AddArgTensorMappings()
532 TF_RETURN_IF_ERROR(add_conversion(map_defun_fn_->arg_nodes[i], false)); in AddArgTensorMappings()
/external/v8/src/wasm/
Dfunction-body-decoder.cc787 TFNode** arg_nodes = builder_->Buffer(param_count + 1); in DoCall() local
789 arg_nodes[0] = index_node; in DoCall()
791 arg_nodes[i + 1] = args[i].node; in DoCall()
794 builder_->CallIndirect(index, arg_nodes, &return_nodes, in DoCall()
797 builder_->CallDirect(index, arg_nodes, &return_nodes, in DoCall()
/external/tensorflow/tensorflow/compiler/jit/
Dextract_outside_compilation_pass.cc75 Status GetArgDataTypes(const std::vector<Node*>& arg_nodes, in GetArgDataTypes() argument
77 recv_at_host_dtypes->resize(arg_nodes.size(), DT_INVALID); in GetArgDataTypes()
78 for (auto* n : arg_nodes) { in GetArgDataTypes()
124 std::vector<Node*> arg_nodes = GatherNodesWithType(*g, "_Arg"); in ReplaceArgNodesWithRecvAtHostNode() local
125 TF_RETURN_IF_ERROR(GetArgDataTypes(arg_nodes, recv_at_host_dtypes)); in ReplaceArgNodesWithRecvAtHostNode()
130 for (auto* n : arg_nodes) { in ReplaceArgNodesWithRecvAtHostNode()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dwhile_op.cc111 const Node* arg = body->arg_nodes[i]; in GetLoopInvariants()