HomeSort by relevance Sort by last modified time
    Searched defs:arg_nodes (Results 1 - 6 of 6) sorted by null

  /external/tensorflow/tensorflow/core/common_runtime/
function.h 80 gtl::InlinedVector<Node*, 4> arg_nodes; member in struct:tensorflow::FunctionBody
partitioning_utils.cc 77 std::vector<std::pair<Node*, int>> arg_nodes; local
89 arg_nodes.push_back(std::make_pair(node, index));
98 for (int i = 0; i < arg_nodes.size(); ++i) {
99 Node* arg = arg_nodes[i].first;
  /external/tensorflow/tensorflow/compiler/tf2xla/
tf2xla.cc 208 Status CollectArgNodes(const Graph& graph, std::vector<Node*>* arg_nodes) {
223 arg_nodes->clear();
225 if (index_node.first != arg_nodes->size()) {
227 arg_nodes->size(), ", but got index ",
230 arg_nodes->push_back(index_node.second);
238 std::vector<Node*> arg_nodes; local
239 TF_RETURN_IF_ERROR(CollectArgNodes(graph, &arg_nodes));
240 for (const Node* node : arg_nodes) {
tf2xla_util.cc 115 std::unordered_map<int, Node*> arg_nodes; local
120 arg_nodes[index] = n;
132 Node* arg_node = arg_nodes[arg_index];
  /external/tensorflow/tensorflow/compiler/jit/
extract_outside_compilation_pass.cc 74 // Gets data types from `arg_nodes` and fills them into `recv_at_host_dtypes`.
75 Status GetArgDataTypes(const std::vector<Node*>& arg_nodes,
77 recv_at_host_dtypes->resize(arg_nodes.size(), DT_INVALID);
78 for (auto* n : arg_nodes) {
124 std::vector<Node*> arg_nodes = GatherNodesWithType(*g, "_Arg"); local
125 TF_RETURN_IF_ERROR(GetArgDataTypes(arg_nodes, recv_at_host_dtypes));
130 for (auto* n : arg_nodes) {
    [all...]
  /external/v8/src/wasm/
function-body-decoder.cc 787 TFNode** arg_nodes = builder_->Buffer(param_count + 1); local
789 arg_nodes[0] = index_node;
791 arg_nodes[i + 1] = args[i].node;
794 builder_->CallIndirect(index, arg_nodes, &return_nodes,
797 builder_->CallDirect(index, arg_nodes, &return_nodes,
    [all...]

Completed in 1485 milliseconds