Searched refs:shape_manager (Results 1 – 1 of 1) sorted by relevance
2096 SymbolicShapeManager* shape_manager) { in ValidateSymbolicShapeManager() argument2111 const auto& merged_shape = shape_manager->GetMergedShape(ic, shape); in ValidateSymbolicShapeManager()2122 const auto& merged_shape = shape_manager->GetMergedShape(ic, shape); in ValidateSymbolicShapeManager()2145 SymbolicShapeManager* shape_manager) { in VerboseShapeInferenceLogging() argument2174 ic->DebugString(shape_manager->GetMergedShape(ic, ic->input(i))), in VerboseShapeInferenceLogging()2180 ic->DebugString(shape_manager->GetMergedShape(ic, ic->output(i))), in VerboseShapeInferenceLogging()2578 std::unique_ptr<SymbolicShapeManager> shape_manager = in InferStatically() local2592 if (!shape_manager->Merge(merged_shapes.first, merged_shapes.second) in InferStatically()2599 if (!shape_manager->Merge(merged_dims.first, merged_dims.second).ok()) { in InferStatically()2607 shape_manager = absl::make_unique<SymbolicShapeManager>(); in InferStatically()[all …]