OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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