Home
last modified time | relevance | path

Searched refs:shape_manager (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc2140 SymbolicShapeManager* shape_manager) { in ValidateSymbolicShapeManager() argument
2155 const auto& merged_shape = shape_manager->GetMergedShape(ic, shape); in ValidateSymbolicShapeManager()
2166 const auto& merged_shape = shape_manager->GetMergedShape(ic, shape); in ValidateSymbolicShapeManager()
2189 SymbolicShapeManager* shape_manager) { in VerboseShapeInferenceLogging() argument
2218 ic->DebugString(shape_manager->GetMergedShape(ic, ic->input(i))), in VerboseShapeInferenceLogging()
2224 ic->DebugString(shape_manager->GetMergedShape(ic, ic->output(i))), in VerboseShapeInferenceLogging()
2622 std::unique_ptr<SymbolicShapeManager> shape_manager = in InferStatically() local
2636 if (!shape_manager->Merge(merged_shapes.first, merged_shapes.second) in InferStatically()
2643 if (!shape_manager->Merge(merged_dims.first, merged_dims.second).ok()) { in InferStatically()
2651 shape_manager = std::make_unique<SymbolicShapeManager>(); in InferStatically()
[all …]