Searched refs:GetInputProperties (Results 1 – 16 of 16) sorted by relevance
109 const auto& input_props = properties.GetInputProperties(node.name()); in Optimize()164 const auto& prop1 = properties.GetInputProperties(input1.node->name()); in Optimize()165 const auto& prop2 = properties.GetInputProperties(input2.node->name()); in Optimize()
404 properties.GetInputProperties(node->name()); in MaterializeShapes()566 properties.GetInputProperties(shape_node.name()); in ExtractShape()737 properties.GetInputProperties(node->name()); in MaterializeReductionIndices()1003 properties->GetInputProperties(node.name()); in IsFoldableUncached()1692 properties.GetInputProperties(node.name()); in IsSimplifiableReshape()2141 const auto& shape = properties.GetInputProperties(node->name())[0].shape(); in RemoveShuffleOrTranspose()2174 !properties.GetInputProperties(node->name()).empty()) { in RemoveRandomShuffle()2175 const auto& shape = properties.GetInputProperties(node->name())[0].shape(); in RemoveRandomShuffle()2193 const auto& shape = properties.GetInputProperties(node->name())[0].shape(); in RemoveReverse()2233 const auto& input = properties.GetInputProperties(node->name())[0]; in SimplifySlice()[all …]
40 properties.GetInputProperties(node.name()); in PredictExecutionTime()
336 ctx.graph_properties.GetInputProperties(contraction_node.name()); in IsCpuCompatibleDataType()658 const auto& props = ctx.graph_properties.GetInputProperties(node.name()); in IsCpuCompatibleDataType()784 const auto& props = ctx.graph_properties.GetInputProperties(node_def->name()); in IsCpuCompatibleDataType()880 const auto& props = ctx.graph_properties.GetInputProperties( in IsCpuCompatibleDataType()934 ctx.graph_properties.GetInputProperties(relu_fanin_0_node_def->name()); in IsCpuCompatibleDataType()
2597 ctx().graph_properties->GetInputProperties(node->name())[0]; in TrySimplify()2717 if (ctx().graph_properties->GetInputProperties(input->name()).size() < 2) { in TrySimplify()2736 ctx().graph_properties->GetInputProperties(add_node->name())[i]; in TrySimplifyInternal()2738 ctx().graph_properties->GetInputProperties(add_node->name())[j]; in TrySimplifyInternal()2809 if (ctx().graph_properties->GetInputProperties(node->name()).size() < 2) { in TrySimplify()2812 const auto& t = ctx().graph_properties->GetInputProperties(node->name())[0]; in TrySimplify()2813 const auto& c = ctx().graph_properties->GetInputProperties(node->name())[1]; in TrySimplify()3193 ctx().graph_properties->GetInputProperties(node->name()); in CheckInputs()
1192 properties.GetInputProperties(node->name()); in SwappingPass()
657 graph_properties_.GetInputProperties(node_->name())[pos].dtype(); in AddLayoutTransposeToInputs()
503 ->GetInputProperties(dst_node->GetName())[dst_port] in UpdateEdge()
166 EXPECT_EQ(1, properties.GetInputProperties(node.name()).size()); in TEST_F()177 const auto in_props = properties.GetInputProperties(node.name()); in TEST_F()206 EXPECT_EQ(1, properties.GetInputProperties(node.name()).size()); in TEST_F()212 const auto in_props = properties.GetInputProperties(node.name()); in TEST_F()215 const auto cleared_props = properties.GetInputProperties(node.name()); in TEST_F()235 EXPECT_EQ(0, properties.GetInputProperties(node.name()).size()); in TEST_F()241 const auto props = properties.GetInputProperties(node.name()); in TEST_F()247 const auto props = properties.GetInputProperties(node.name()); in TEST_F()577 const auto props = properties.GetInputProperties("Less"); in TEST_F()905 const auto input_props = properties.GetInputProperties("init_restore"); in TEST_F()[all …]
140 const std::vector<OpInfo::TensorProperties>& GetInputProperties(
90 properties->GetInputProperties(node.name()); in InferMemUsageForNodes()
789 graph_properties_->GetInputProperties(node->name()); in GetNodeStateOrCreateIt()
2791 GraphProperties::GetInputProperties(const string& node_name) const { in GetInputProperties() function in tensorflow::grappler::GraphProperties
90 properties.GetInputProperties(node->name()); in PrintNodeInfo()
519 graph_properties->GetInputProperties(node->name())); in OperationCanBeTranslatedToImplicitBatch()522 graph_properties->GetInputProperties(node->name()); in OperationCanBeTranslatedToImplicitBatch()657 graph_properties->GetInputProperties(node->name()); in GetClusterBatchSizeForNode()
280 void GetInputProperties(const grappler::GraphProperties& graph_properties, in GetInputProperties() function284 auto input_params = graph_properties.GetInputProperties(node->name()); in GetInputProperties()6480 GetInputProperties(graph_properties, in ConvertSegmentToGraphDef()