Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/costs/
Dvirtual_scheduler.cc564 port_num_end = curr_node_state.output_properties.size(); in Init()
599 auto& outputs = node_state.output_properties; in MaybeUpdateInputOutput()
621 const auto& output_properties = in MaybeUpdateInputOutput() local
625 if (!output_properties.empty()) { in MaybeUpdateInputOutput()
629 CHECK_GT(output_properties.size(), input_node_port_num); in MaybeUpdateInputOutput()
630 inputs.push_back(output_properties[input_node_port_num]); in MaybeUpdateInputOutput()
631 outputs.push_back(output_properties[input_node_port_num]); in MaybeUpdateInputOutput()
766 for (auto& output : node_state.output_properties) { in CreateOpContext()
790 node_state.output_properties = in GetNodeStateOrCreateIt()
807 for (size_t i = 0; i < node_state.output_properties.size(); ++i) { in GetNodeStateOrCreateIt()
[all …]
Dutils.cc224 const std::vector<OpInfo::TensorProperties>& output_properties, in CalculateOutputSize() argument
228 if (port_num >= output_properties.size()) { in CalculateOutputSize()
230 << " >= output_properties.size(): " << output_properties.size(); in CalculateOutputSize()
234 return CalculateTensorSize(output_properties[port_num]); in CalculateOutputSize()
Dgraph_properties.cc317 const auto& output_properties = output_properties_map.at(node.name()); in VerboseLogUnknownDimensionSources() local
331 for (const auto& output_prop : output_properties) { in VerboseLogUnknownDimensionSources()
340 for (const auto& output_prop : output_properties) { in VerboseLogUnknownDimensionSources()
932 auto output_properties = gp.GetOutputProperties(retnode->name()); in UpdateFunction() local
933 int output_properties_size = output_properties.size(); in UpdateFunction()
937 " (output_properties.size() = ", output_properties.size(), ")."); in UpdateFunction()
939 auto& outprop = output_properties[out_tensor.index()]; in UpdateFunction()
2662 auto& output_properties = output_properties_[node.name()]; in InferStatically() local
2665 CHECK_EQ(output_properties.size(), 0); in InferStatically()
2667 output_properties.resize(ic->num_outputs()); in InferStatically()
[all …]
Dutils.h55 const std::vector<OpInfo::TensorProperties>& output_properties,
Dvirtual_scheduler.h54 std::vector<OpInfo::TensorProperties> output_properties; member
Dvirtual_scheduler_test.cc2993 std::vector<OpInfo::TensorProperties> output_properties; in TEST_F() local
2995 output_properties.push_back(output_property); in TEST_F()
2997 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 int num_outputs = output_properties.size(); in GetTensorProperties()
70 *properties = &output_properties[tensor_id.index()]; in GetTensorProperties()
Dpin_to_host_optimizer.cc109 const auto& output_properties = properties->GetOutputProperties(node.name()); in IsNodeOutputPortHostFriendly() local
110 int output_properties_size = output_properties.size(); in IsNodeOutputPortHostFriendly()
113 << " but output_properties.size()=" << output_properties.size() in IsNodeOutputPortHostFriendly()
118 if (!IsTensorSmall(output_properties[port_id])) { in IsNodeOutputPortHostFriendly()
Darithmetic_optimizer.cc3458 const OpInfo::TensorProperties* output_properties; in RewriteGraph() local
3460 strings::StrCat(node->name(), ":", 0), &output_properties)); in RewriteGraph()
3461 PartialTensorShape output_shape(output_properties->shape()); in RewriteGraph()
3467 SetDataTypeToAttr(output_properties->dtype(), "T", output); in RewriteGraph()
3486 SetDataTypeToAttr(output_properties->dtype(), "T", output); in RewriteGraph()