OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
defs:ret_nodes
(Results
1 - 3
of
3
) sorted by null
/external/tensorflow/tensorflow/core/common_runtime/
function.h
81
gtl::InlinedVector<Node*, 4>
ret_nodes
;
member in struct:tensorflow::FunctionBody
partitioning_utils.cc
78
std::vector<std::pair<Node*, int>>
ret_nodes
;
local
94
ret_nodes
.push_back(std::make_pair(node, index));
109
for (int i = 0; i <
ret_nodes
.size(); ++i) {
110
Node* ret =
ret_nodes
[i].first;
/external/tensorflow/tensorflow/compiler/jit/
extract_outside_compilation_pass.cc
175
// Gets data types from `
ret_nodes
` and fills them into `send_from_host_dtypes`.
176
Status GetRetDataTypes(const std::vector<Node*>&
ret_nodes
,
178
send_from_host_dtypes->resize(
ret_nodes
.size(), DT_INVALID);
179
for (auto* n :
ret_nodes
) {
197
const std::vector<Node*>&
ret_nodes
,
213
for (auto* n :
ret_nodes
) {
240
std::vector<Node*>
ret_nodes
= GatherNodesWithType(*g, "_Retval");
local
241
TF_RETURN_IF_ERROR(GetRetDataTypes(
ret_nodes
, send_from_host_dtypes));
244
BuildSendFromHostNode(g, oc_cluster_name,
ret_nodes
,
246
for (auto* n :
ret_nodes
) {
[
all
...]
Completed in 269 milliseconds