OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:output_properties
(Results
1 - 10
of
10
) sorted by null
/external/tensorflow/tensorflow/core/grappler/optimizers/
graph_optimizer_stage.cc
59
const auto&
output_properties
=
local
61
auto num_outputs =
output_properties
.size();
70
properties->CopyFrom(
output_properties
[tensor_id.index()]);
pin_to_host_optimizer.cc
106
const auto&
output_properties
= properties->GetOutputProperties(node.name());
local
107
if (port_id >=
output_properties
.size()) {
109
<< " but
output_properties
.size()=" <<
output_properties
.size()
114
if (!IsTensorSmall(
output_properties
[port_id])) {
arithmetic_optimizer.cc
3266
OpInfo::TensorProperties
output_properties
;
local
[
all
...]
/external/tensorflow/tensorflow/core/grappler/costs/
virtual_scheduler.cc
443
port_num < curr_node_state.
output_properties
.size(); ++port_num) {
477
auto& outputs = node_state.
output_properties
;
499
const auto&
output_properties
=
local
503
if (!
output_properties
.empty()) {
507
CHECK_GT(
output_properties
.size(), input_node_port_num);
508
inputs.push_back(
output_properties
[input_node_port_num]);
509
outputs.push_back(
output_properties
[input_node_port_num]);
641
for (auto& output : node_state.
output_properties
) {
665
node_state.
output_properties
=
680
for (size_t i = 0; i < node_state.
output_properties
.size(); ++i)
[
all
...]
utils.h
54
const std::vector<OpInfo::TensorProperties>&
output_properties
,
graph_properties.cc
310
const auto&
output_properties
= output_properties_map.at(node.name());
local
324
for (const auto& output_prop :
output_properties
) {
333
for (const auto& output_prop :
output_properties
) {
715
auto
output_properties
= gp.GetOutputProperties(retnode->name());
local
716
if (out_tensor.index() >=
output_properties
.size()) {
719
" (
output_properties
.size() = ",
output_properties
.size(), ").");
721
auto const& outprop =
output_properties
[out_tensor.index()];
2218
auto&
output_properties
=
output_properties
_[node.name()];
local
2287
std::vector<OpInfo::TensorProperties>
output_properties
;
local
[
all
...]
utils.cc
223
const std::vector<OpInfo::TensorProperties>&
output_properties
,
227
if (port_num >=
output_properties
.size()) {
229
<< " >=
output_properties
.size(): " <<
output_properties
.size();
233
return CalculateTensorSize(
output_properties
[port_num]);
virtual_scheduler.h
47
std::vector<OpInfo::TensorProperties>
output_properties
;
member in struct:tensorflow::grappler::NodeState
65
// They should be 1 +
output_properties
.size() as we add [-1] for control
virtual_scheduler_test.cc
[
all
...]
/external/tensorflow/tensorflow/python/grappler/
hierarchical_controller.py
649
for i,
output_properties
in enumerate(self.node_properties[op.name]):
650
if
output_properties
.shape.__str__() == "<unknown>":
654
shape =
output_properties
.shape
[
all
...]
Completed in 411 milliseconds