Searched refs:GetOutputProperties (Results 1 – 17 of 17) sorted by relevance
137 const auto props = properties.GetOutputProperties(node.name()); in TEST_F()154 const auto out_props = properties.GetOutputProperties(node.name()); in TEST_F()179 const auto props = properties.GetOutputProperties(node.name()); in TEST_F()181 const auto cleared_props = properties.GetOutputProperties(node.name()); in TEST_F()227 const auto out_props = properties.GetOutputProperties(node.name()); in TEST_F()264 const auto props = static_properties.GetOutputProperties("Var"); in TEST_F()278 const auto props = dynamic_properties.GetOutputProperties("Var"); in TEST_F()309 const auto props = properties.GetOutputProperties("ReadVariableOpAfterEnter"); in TEST_F()334 const auto props = properties.GetOutputProperties("VarRead"); in TEST_F()371 const auto props = properties.GetOutputProperties(node); in TEST_F()[all …]
118 const std::vector<OpInfo::TensorProperties>& GetOutputProperties(
81 properties->GetOutputProperties(node.name()); in InferMemUsageForNodes()
500 graph_properties_->GetOutputProperties(NodeName(input_source_name)); in MaybeUpdateInputOutput()666 graph_properties_->GetOutputProperties(node->name()); in GetNodeStateOrCreateIt()
715 auto output_properties = gp.GetOutputProperties(retnode->name()); in UpdateFunction()2269 auto tensor_properties = GetOutputProperties(node->name()); in AnnotateOutputShapes()2332 GraphProperties::GetOutputProperties(const string& node_name) const { in GetOutputProperties() function in tensorflow::grappler::GraphProperties
106 const auto& output_properties = properties->GetOutputProperties(node.name()); in IsNodeOutputPortHostFriendly()256 for (const auto& prop : properties->GetOutputProperties(node.name())) { in IsNodeHostCandidate()
60 ctx.graph_properties->GetOutputProperties(tensor_id.node()); in GetTensorProperties()
64 properties.GetOutputProperties(reduce_indices.node->name()); in Optimize()
44 properties.GetOutputProperties(node.name()); in PredictExecutionTime()
359 properties.GetOutputProperties(node->name()); in MaterializeShapes()705 properties.GetOutputProperties(node->name()); in MaterializeReductionIndices()725 properties.GetOutputProperties(fanout->name()); in MaterializeReductionIndices()779 properties.GetOutputProperties(node->name()); in MaterializeConstantValuedNode()1019 auto output_props = properties.GetOutputProperties(node.name()); in GetDataTypeFromNodeOrProps()2484 const auto& output_props = properties.GetOutputProperties(node.name())[0]; in IsReductionCandidateForSimplification()2659 properties.GetOutputProperties(node->name())[0].shape(); in SimplifyArithmeticOperations()2946 const auto& mul_props = properties.GetOutputProperties(node->name()); in MulConvPushDown()2947 const auto& conv_props = properties.GetOutputProperties(conv_node->name()); in MulConvPushDown()2965 const auto& const_props = properties.GetOutputProperties(const_node->name()); in MulConvPushDown()
94 graph_properties.GetOutputProperties(n->name()); in CheckTypesAndGetShapes()
585 properties.GetOutputProperties(node->name())[0].shape(); in SchedulingPass()
683 graph_properties_.GetOutputProperties(node_->name())[input_port] in AddTransformToOutputs()
2503 ctx().graph_properties->GetOutputProperties(node->name())[0].shape(); in TrySimplify()
50 properties.GetOutputProperties(node->name()); in PrintNodeInfo()
158 properties.GetOutputProperties(node_name); in TF_GetOpProperties()
233 void GetOutputProperties(const grappler::GraphProperties& graph_properties, in GetOutputProperties() function237 auto output_params = graph_properties.GetOutputProperties(node->name()); in GetOutputProperties()917 graph_properties.GetOutputProperties(node_def.name()); in ConvertToTensorOrWeights()4189 GetOutputProperties(graph_properties, in ConvertSegmentToGraphDef()