OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:input_node_info
(Results
1 - 4
of
4
) sorted by null
/external/tensorflow/tensorflow/core/kernels/hexagon/
graph_transfer_utils.cc
115
for (const std::pair<string, Tensor>&
input_node_info
: inputs) {
116
const Scope& scope = root.WithOpName(
input_node_info
.first);
120
.Attr("dtype",
input_node_info
.second.dtype())
121
.Attr("shape",
input_node_info
.second.shape());
126
input_types.push_back(
input_node_info
.second.dtype());
graph_transferer.cc
143
for (const std::pair<string, Tensor>&
input_node_info
:
147
graph_input_node_info.set_name(
input_node_info
.first);
148
graph_input_node_info.set_dtype(
input_node_info
.second.dtype());
149
for (const int64 dim : ToTensorShapeArray(
input_node_info
.second.shape())) {
[
all
...]
/external/tensorflow/tensorflow/core/kernels/
remote_fused_graph_execute_utils_test.cc
228
std::pair<string, Tensor>
input_node_info
;
local
229
input_node_info
.first = NAME_A;
230
input_node_info
.second = Tensor(DT_FLOAT, {});
231
input_node_info
.second.scalar<float>()() = 1.0f;
232
const std::vector<std::pair<string, Tensor>> inputs{
input_node_info
};
249
std::pair<string, Tensor>
input_node_info
;
local
250
input_node_info
.first = NAME_A;
251
input_node_info
.second = Tensor(DT_FLOAT, {});
252
const std::vector<std::pair<string, Tensor>> inputs{
input_node_info
};
[
all
...]
remote_fused_graph_execute_utils.cc
302
for (const std::pair<string, Tensor>&
input_node_info
:
304
output_tensors.push_back(
input_node_info
.second);
451
for (const std::pair<string, Tensor>&
input_node_info
:
453
if (node->name() ==
input_node_info
.first) {
458
input_node_info
.second.shape(), &handle);
[
all
...]
Completed in 552 milliseconds