Searched refs:GetInputProperties (Results 1 – 14 of 14) sorted by relevance
135 EXPECT_EQ(1, properties.GetInputProperties(node.name()).size()); in TEST_F()146 const auto in_props = properties.GetInputProperties(node.name()); in TEST_F()178 EXPECT_EQ(1, properties.GetInputProperties(node.name()).size()); in TEST_F()184 const auto in_props = properties.GetInputProperties(node.name()); in TEST_F()187 const auto cleared_props = properties.GetInputProperties(node.name()); in TEST_F()207 EXPECT_EQ(0, properties.GetInputProperties(node.name()).size()); in TEST_F()213 const auto props = properties.GetInputProperties(node.name()); in TEST_F()219 const auto props = properties.GetInputProperties(node.name()); in TEST_F()539 const auto props = properties.GetInputProperties("Less"); in TEST_F()867 const auto input_props = properties.GetInputProperties("init_restore"); in TEST_F()[all …]
116 const std::vector<OpInfo::TensorProperties>& GetInputProperties(
91 properties->GetInputProperties(node.name()); in InferMemUsageForNodes()
664 graph_properties_->GetInputProperties(node->name()); in GetNodeStateOrCreateIt()
2323 GraphProperties::GetInputProperties(const string& node_name) const { in GetInputProperties() function in tensorflow::grappler::GraphProperties
71 const auto& input_props = properties.GetInputProperties(node.name()); in Optimize()105 const auto& prop1 = properties.GetInputProperties(input1.node->name()); in Optimize()106 const auto& prop2 = properties.GetInputProperties(input2.node->name()); in Optimize()
148 properties.GetInputProperties(node->name()).empty()) { in GetConcatAxis()151 const auto& axis_input = properties.GetInputProperties(node->name()).back(); in GetConcatAxis()361 properties.GetInputProperties(node->name()); in MaterializeShapes()512 properties.GetInputProperties(shape_node.name()); in ExtractShape()682 properties.GetInputProperties(node->name()); in MaterializeReductionIndices()1522 properties.GetInputProperties(node.name()); in IsSimplifiableReshape()1938 properties.GetInputProperties(node->name()).size() >= 2) { in RemoveShuffleOrTranspose()1939 const auto& shape = properties.GetInputProperties(node->name())[0].shape(); in RemoveShuffleOrTranspose()1944 const auto& p = properties.GetInputProperties(node->name())[1]; in RemoveShuffleOrTranspose()1985 !properties.GetInputProperties(node->name()).empty()) { in RemoveRandomShuffle()[all …]
38 properties.GetInputProperties(node.name()); in PredictExecutionTime()
2467 ctx().graph_properties->GetInputProperties(node->name()).size() == 2; in IsSupported()2472 ctx().graph_properties->GetInputProperties(node->name())[1]; in TrySimplify()2501 ctx().graph_properties->GetInputProperties(node->name())[0]; in TrySimplify()2609 if (ctx().graph_properties->GetInputProperties(input->name()).size() < 2) { in TrySimplify()2628 ctx().graph_properties->GetInputProperties(input->name())[i]; in TrySimplifyInternal()2630 ctx().graph_properties->GetInputProperties(input->name())[j]; in TrySimplifyInternal()2701 if (ctx().graph_properties->GetInputProperties(node->name()).size() < 2) { in TrySimplify()2711 if (ctx().graph_properties->GetInputProperties(exp->name()).empty()) { in TrySimplify()2715 const auto& t = ctx().graph_properties->GetInputProperties(exp->name())[0]; in TrySimplify()2716 const auto& c = ctx().graph_properties->GetInputProperties(node->name())[1]; in TrySimplify()[all …]
199 ctx.graph_properties.GetInputProperties(matched.conv2d->name()); in IsGpuCompatible()437 const auto& props = ctx.graph_properties.GetInputProperties(node->name()); in FindFusedBatchNorm()
1152 properties.GetInputProperties(node->name()); in SwappingPass()
645 graph_properties_.GetInputProperties(node_->name())[pos].dtype(); in AddLayoutTransposeToInputs()
90 properties.GetInputProperties(node->name()); in PrintNodeInfo()
247 void GetInputProperties(const grappler::GraphProperties& graph_properties, in GetInputProperties() function251 auto input_params = graph_properties.GetInputProperties(node->name()); in GetInputProperties()4194 GetInputProperties(graph_properties, in ConvertSegmentToGraphDef()