Home
last modified time | relevance | path

Searched refs:output_properties (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/costs/
Dvirtual_scheduler.cc443 port_num < curr_node_state.output_properties.size(); ++port_num) { in Init()
477 auto& outputs = node_state.output_properties; in MaybeUpdateInputOutput()
499 const auto& output_properties = in MaybeUpdateInputOutput() local
503 if (!output_properties.empty()) { in MaybeUpdateInputOutput()
507 CHECK_GT(output_properties.size(), input_node_port_num); in MaybeUpdateInputOutput()
508 inputs.push_back(output_properties[input_node_port_num]); in MaybeUpdateInputOutput()
509 outputs.push_back(output_properties[input_node_port_num]); in MaybeUpdateInputOutput()
641 for (auto& output : node_state.output_properties) { in GetCurrNode()
665 node_state.output_properties = in GetNodeStateOrCreateIt()
680 for (size_t i = 0; i < node_state.output_properties.size(); ++i) { in GetNodeStateOrCreateIt()
[all …]
Dgraph_properties.cc310 const auto& output_properties = output_properties_map.at(node.name()); in VerboseLogUnknownDimensionSources() local
324 for (const auto& output_prop : output_properties) { in VerboseLogUnknownDimensionSources()
333 for (const auto& output_prop : output_properties) { in VerboseLogUnknownDimensionSources()
715 auto output_properties = gp.GetOutputProperties(retnode->name()); in UpdateFunction() local
716 if (out_tensor.index() >= output_properties.size()) { in UpdateFunction()
719 " (output_properties.size() = ", output_properties.size(), ")."); in UpdateFunction()
721 auto const& outprop = output_properties[out_tensor.index()]; in UpdateFunction()
2218 auto& output_properties = output_properties_[node.name()]; in InferStatically() local
2221 CHECK_EQ(output_properties.size(), 0); in InferStatically()
2223 output_properties.resize(ic->num_outputs()); in InferStatically()
[all …]
Dutils.cc223 const std::vector<OpInfo::TensorProperties>& output_properties, in CalculateOutputSize() argument
227 if (port_num >= output_properties.size()) { in CalculateOutputSize()
229 << " >= output_properties.size(): " << output_properties.size(); in CalculateOutputSize()
233 return CalculateTensorSize(output_properties[port_num]); in CalculateOutputSize()
Dutils.h54 const std::vector<OpInfo::TensorProperties>& output_properties,
Dvirtual_scheduler.h47 std::vector<OpInfo::TensorProperties> output_properties; member
Dvirtual_scheduler_test.cc2816 std::vector<OpInfo::TensorProperties> output_properties; in TEST_F() local
2818 output_properties.push_back(output_property); in TEST_F()
2820 return CalculateOutputSize(output_properties, 0); in TEST_F()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgraph_optimizer_stage.cc59 const auto& output_properties = in GetTensorProperties() local
61 auto num_outputs = output_properties.size(); in GetTensorProperties()
70 properties->CopyFrom(output_properties[tensor_id.index()]); in GetTensorProperties()
Dpin_to_host_optimizer.cc106 const auto& output_properties = properties->GetOutputProperties(node.name()); in IsNodeOutputPortHostFriendly() local
107 if (port_id >= output_properties.size()) { in IsNodeOutputPortHostFriendly()
109 << " but output_properties.size()=" << output_properties.size() in IsNodeOutputPortHostFriendly()
114 if (!IsTensorSmall(output_properties[port_id])) { in IsNodeOutputPortHostFriendly()
Darithmetic_optimizer.cc3266 OpInfo::TensorProperties output_properties; in RewriteGraph() local
3268 strings::StrCat(node->name(), ":", 0), &output_properties)); in RewriteGraph()
3269 PartialTensorShape output_shape(output_properties.shape()); in RewriteGraph()
3275 SetDataTypeToAttr(output_properties.dtype(), "T", output); in RewriteGraph()
3290 SetDataTypeToAttr(output_properties.dtype(), "T", output); in RewriteGraph()
/external/tensorflow/tensorflow/python/grappler/
Dhierarchical_controller.py649 for i, output_properties in enumerate(self.node_properties[op.name]):
650 if output_properties.shape.__str__() == "<unknown>":
654 shape = output_properties.shape