Searched refs:GetOutputProperties (Results 1 – 20 of 20) sorted by relevance
168 const auto props = properties.GetOutputProperties(node.name()); in TEST_F()185 const auto out_props = properties.GetOutputProperties(node.name()); in TEST_F()207 const auto props = properties.GetOutputProperties(node.name()); in TEST_F()209 const auto cleared_props = properties.GetOutputProperties(node.name()); in TEST_F()255 const auto out_props = properties.GetOutputProperties(node.name()); in TEST_F()302 const auto props = static_properties.GetOutputProperties("Var"); in TEST_F()316 const auto props = dynamic_properties.GetOutputProperties("Var"); in TEST_F()347 const auto props = properties.GetOutputProperties("ReadVariableOpAfterEnter"); in TEST_F()372 const auto props = properties.GetOutputProperties("VarRead"); in TEST_F()409 const auto props = properties.GetOutputProperties(node); in TEST_F()[all …]
142 const std::vector<OpInfo::TensorProperties>& GetOutputProperties(
80 properties->GetOutputProperties(node.name()); in InferMemUsageForNodes()
622 graph_properties_->GetOutputProperties(NodeName(input_source_name)); in MaybeUpdateInputOutput()791 graph_properties_->GetOutputProperties(node->name()); in GetNodeStateOrCreateIt()
932 auto output_properties = gp.GetOutputProperties(retnode->name()); in UpdateFunction()2733 auto tensor_properties = GetOutputProperties(node->name()); in AnnotateOutputShapes()2800 GraphProperties::GetOutputProperties(const string& node_name) const { in GetOutputProperties() function in tensorflow::grappler::GraphProperties
109 const auto& output_properties = properties->GetOutputProperties(node.name()); in IsNodeOutputPortHostFriendly()261 for (const auto& prop : properties->GetOutputProperties(node.name())) { in IsNodeHostCandidate()
60 ctx.graph_properties->GetOutputProperties(tensor_id.node()); in GetTensorProperties()
101 properties.GetOutputProperties(reduce_indices.node->name()); in Optimize()
402 properties.GetOutputProperties(node->name()); in MaterializeShapes()760 properties.GetOutputProperties(node->name()); in MaterializeReductionIndices()780 properties.GetOutputProperties(fanout->name()); in MaterializeReductionIndices()837 properties.GetOutputProperties(node->name()); in MaterializeConstantValuedNode()900 properties.GetOutputProperties(node->name()); in MaterializeOutputValues()1005 properties->GetOutputProperties(node.name()); in IsFoldableUncached()1178 auto output_props = properties.GetOutputProperties(node.name()); in GetDataTypeFromNodeOrProps()1819 properties.GetOutputProperties(node->name())[0].shape()); in ReplaceOperationWithBroadcastTo()2754 const auto& output_props = properties.GetOutputProperties(node.name())[0]; in IsReductionCandidateForSimplification()2939 properties.GetOutputProperties(node->name())[0].shape(); in SimplifyArithmeticOperations()[all …]
46 properties.GetOutputProperties(node.name()); in PredictExecutionTime()
110 graph_properties.GetOutputProperties(n->name()); in CheckTypesAndGetShapes()300 graph_properties.GetOutputProperties(inode->name()); in GetInputs()
606 properties.GetOutputProperties(node->name())[0].shape(); in SchedulingPass()
695 graph_properties_.GetOutputProperties(node_->name())[input_port] in AddTransformToOutputs()
506 ->GetOutputProperties(src_node->GetName())[src_port] in UpdateEdge()
2599 ctx().graph_properties->GetOutputProperties(node->name())[0].shape(); in TrySimplify()
50 properties.GetOutputProperties(node->name()); in PrintNodeInfo()
201 output_props = properties.GetOutputProperties(node_name); in PYBIND11_MODULE()
569 graph_properties->GetOutputProperties(node->name())); in OperationHasDynamicNonBatchDimension()571 graph_properties->GetOutputProperties(node->name()).at(0)); in OperationHasDynamicNonBatchDimension()
270 tensorflow::grappler::GraphProperties::GetOutputProperties
266 void GetOutputProperties(const grappler::GraphProperties& graph_properties, in GetOutputProperties() function270 auto output_params = graph_properties.GetOutputProperties(node->name()); in GetOutputProperties()1170 graph_properties_.GetOutputProperties(node_def.name()); in ConvertToTensorOrWeights()6475 GetOutputProperties(graph_properties, in ConvertSegmentToGraphDef()