Searched refs:output_properties (Results 1 – 9 of 9) sorted by relevance
/external/tensorflow/tensorflow/core/grappler/costs/ |
D | virtual_scheduler.cc | 564 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 …]
|
D | utils.cc | 224 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()
|
D | graph_properties.cc | 317 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 …]
|
D | utils.h | 55 const std::vector<OpInfo::TensorProperties>& output_properties,
|
D | virtual_scheduler.h | 54 std::vector<OpInfo::TensorProperties> output_properties; member
|
D | virtual_scheduler_test.cc | 2993 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/ |
D | graph_optimizer_stage.cc | 59 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()
|
D | pin_to_host_optimizer.cc | 109 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()
|
D | arithmetic_optimizer.cc | 3458 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()
|