HomeSort by relevance Sort by last modified time
    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